結果
問題 | 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 = 998244353D = 33N, M = map(int, input().split())dp = [[[0] * (D + 1) for _ in range(2)] for _ in range(D + 1)]dp[D][0][0] = 1for i in reversed(range(D)):nd = (M >> i) & 1for j in range(2):for k in range(D):for d in range(2):nj = jif nj == 0:if d > nd:continueif d < nd:nj = 1dp[i][nj][k+d] += dp[i+1][j][k]ans = 0for i in range(D):ans += pow(N, i, mod) * (dp[0][0][i] + dp[0][1][i]) % modans %= modprint(ans)