結果

問題 No.1761 Sequence Distance
ユーザー NyaanNyaan
提出日時 2021-10-24 22:47:16
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 7,215 ms / 8,000 ms
コード長 728 bytes
コンパイル時間 778 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 78,720 KB
最終ジャッジ日時 2025-01-03 13:24:15
合計ジャッジ時間 66,343 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import sqrt
from copy import deepcopy

mod = 998244353
N, M = map(int, input().split())
sq = int(sqrt(2 * M) + 3)

init = [[0] * (M + 1) for _ in range(sq)]
dp = deepcopy(init)
nx = deepcopy(init)
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])
0