結果

問題 No.1761 Sequence Distance
ユーザー NyaanNyaan
提出日時 2021-10-24 22:56:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 7,424 ms / 8,000 ms
コード長 673 bytes
コンパイル時間 509 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 77,620 KB
最終ジャッジ日時 2025-01-03 13:30:53
合計ジャッジ時間 71,818 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

from math import sqrt

mod = 998244353
N, M = map(int, input().split())

dp = [[0] * (M + 1) for _ in range(50)]
nx = [[0] * (M + 1) for _ in range(50)]
dp[0][0] = 1

for _ in range(N):
  for i in range(50):
    for j in range(M + 1):
      nx[i][j] = 0
  for i in range(50):
    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 != 50 - 1 and j + i + 1 <= M:
        nx[i + 1][j + i + 1] += dp[i][j]
  dp, nx = nx, dp
  for i in range(50):
    for j in range(M + 1):
      dp[i][j] %= mod

print(dp[0][M])
0