結果
問題 | No.2060 AND Sequence |
ユーザー |
![]() |
提出日時 | 2022-08-26 22:38:40 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 33 ms / 2,000 ms |
コード長 | 613 bytes |
コンパイル時間 | 400 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 10,752 KB |
最終ジャッジ日時 | 2024-10-13 23:16:57 |
合計ジャッジ時間 | 3,310 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 |
ソースコード
mod = 998244353 D = 33 N, M = map(int, input().split()) dp = [[[0] * (D + 1) for _ in range(2)] for _ in range(D + 1)] dp[D][0][0] = 1 for i in reversed(range(D)): nd = (M >> i) & 1 for j in range(2): for k in range(D): for d in range(2): nj = j if nj == 0: if d > nd: continue if d < nd: nj = 1 dp[i][nj][k+d] += dp[i+1][j][k] ans = 0 for i in range(D): ans += pow(N, i, mod) * (dp[0][0][i] + dp[0][1][i]) % mod ans %= mod print(ans)