結果
問題 |
No.1761 Sequence Distance
|
ユーザー |
|
提出日時 | 2021-10-24 22:53:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 5,919 ms / 8,000 ms |
コード長 | 695 bytes |
コンパイル時間 | 370 ms |
コンパイル使用メモリ | 82,508 KB |
実行使用メモリ | 77,516 KB |
最終ジャッジ日時 | 2025-01-03 13:21:55 |
合計ジャッジ時間 | 55,356 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
from math import sqrt mod = 998244353 N, M = map(int, input().split()) sq = int(sqrt(M) + 3) dp = [[0] * (M + 1) for _ in range(sq)] nx = [[0] * (M + 1) for _ in range(sq)] dp[0][0] = 1 for _ in range(N): for i in range(sq): for j in range(M + 1): nx[i][j] = 0 for i in range(sq): for j in range(M + 1): if dp[i][j] == 0: continue im1 = abs(i - 1) if j + im1 <= M: nx[im1][j + im1] += dp[i][j] if j + i <= M: nx[i][j + i] += dp[i][j] * 2 if i != sq - 1 and j + i + 1 <= M: nx[i + 1][j + i + 1] += dp[i][j] dp, nx = nx, dp for i in range(sq): for j in range(M + 1): dp[i][j] %= mod print(dp[0][M])