結果

問題 No.1761 Sequence Distance
ユーザー NyaanNyaan
提出日時 2021-10-24 22:49:13
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 749 bytes
コンパイル時間 436 ms
コンパイル使用メモリ 82,660 KB
実行使用メモリ 79,232 KB
最終ジャッジ日時 2025-01-03 13:19:28
合計ジャッジ時間 74,769 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40 TLE * 2
権限があれば一括ダウンロードができます

ソースコード

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):
      if dp[i][j]:
        dp[i][j] %= mod

print(dp[0][M])
0