結果

問題 No.3139 Interval MEX ?
コンテスト
ユーザー tikisan_
提出日時 2025-05-20 16:42:36
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 688 bytes
コンパイル時間 823 ms
コンパイル使用メモリ 12,160 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2025-05-20 16:42:46
合計ジャッジ時間 9,694 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other AC * 1 WA * 25 TLE * 1 -- * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve():
    N, M = map(int, input().split())
    MOD = 998244353
    for x in range(N + 1):
        dp = [[0] * (x + 1) for _ in range(N + 1)]
        dp[0][0] = 1
        for i in range(N):
            for j in range(x + 1):
                if dp[i][j] == 0:
                    continue
                max_val = min(i + M, x)
                if j < x:
                    ways = max(0, max_val - j)
                    dp[i + 1][j + 1] = (dp[i + 1][j + 1] + dp[i][j] * ways) % MOD
                ways = max(0, max_val - j + 1)
                dp[i + 1][j] = (dp[i + 1][j] + dp[i][j] * ways) % MOD
        ans = dp[N][x]
        print(ans)

if __name__ == "__main__":
    solve()
0