結果

問題 No.1761 Sequence Distance
コンテスト
ユーザー NyaanNyaan
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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])
0