結果

問題 No.2388 At Least K-Characters
コンテスト
ユーザー kakel-san
提出日時 2023-07-21 22:24:14
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 1,356 ms / 4,000 ms
コード長 1,964 bytes
コンパイル時間 2,531 ms
コンパイル使用メモリ 113,368 KB
実行使用メモリ 156,604 KB
最終ジャッジ日時 2024-07-05 03:46:48
合計ジャッジ時間 30,782 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

using System;
using static System.Console;
using System.Linq;
using System.Collections.Generic;
class Program
{
    static int NN => int.Parse(ReadLine());
    static int[] NList => ReadLine().Split().Select(int.Parse).ToArray();
    public static void Main()
    {
        Solve();
    }
    static void Solve()
    {
        var c = NList;
        var (n, m, k) = (c[0], c[1], c[2]);
        var s = ReadLine();
        var kinds = 26;
        WriteLine(Least(n, m, k, kinds, s));
    }
    static long Least(int n, int m, int k, int kinds, string s)
    {
        var ans = 0L;
        var mod = 998_244_353;
        var dp = new long[m][];
        for (var i = 0; i < dp.Length; ++i) dp[i] = new long[kinds + 1];
        var count = new int[kinds];
        for (var i = 0; i < m; ++i)
        {
            if (i == 0)
            {
                for (var si = 'a'; si < s[0]; ++si) ++dp[0][1];
            }
            else
            {
                for (var j = 1; j < dp[i].Length; ++j)
                {
                    dp[i][j] = (dp[i][j] + dp[i - 1][j] * j) % mod;
                    if (j + 1 < dp[i].Length) dp[i][j + 1] = (dp[i][j + 1] + dp[i - 1][j] * (kinds - j));
                }
                if (i < n)
                {
                    var kind = count.Count(ci => ci > 0);
                    for (var j = 0; j < s[i] - 'a'; ++j)
                    {
                        if (count[j] == 0) ++kind;
                        dp[i][kind] = (dp[i][kind] + 1) % mod;
                        if (count[j] == 0) --kind;
                    }
                }
            }
            if (i < n)
            {
                ++count[s[i] - 'a'];
            }
            if (i + 1 < n && count.Count(ci => ci > 0) >= k) ans = (ans + 1) % mod;
            for (var j = k; j < dp[i].Length; ++j) ans = (ans + dp[i][j]) % mod;
        }
        return ans;
    }
}
0