結果
| 問題 |
No.3213 depth max K
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2025-07-27 02:51:01 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 817 ms / 2,000 ms |
| コード長 | 720 bytes |
| コンパイル時間 | 227 ms |
| コンパイル使用メモリ | 82,652 KB |
| 実行使用メモリ | 77,556 KB |
| 最終ジャッジ日時 | 2025-07-27 02:51:11 |
| 合計ジャッジ時間 | 9,513 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 41 |
ソースコード
# https://yukicoder.me/problems/no/3213s
MOD = 998244353
def main():
N, K = map(int, input().split())
# 深さがK以下のもの
def calc(N, K):
dp = [0] * (K + 1)
dp[0] = 1
for _ in range(2 * N):
new_dp = [0] * (K + 1)
for k in range(K + 1):
if k - 1 >= 0:
new_dp[k - 1] += dp[k]
new_dp[k - 1] %= MOD
if k + 1 <= K:
new_dp[k + 1] += dp[k]
new_dp[k + 1] %= MOD
dp = new_dp
return dp[0]
d1 = calc(N, K)
d2 = calc(N, K -1)
print((d1 - d2) % MOD)
if __name__ == "__main__":
main()