結果
問題 | No.2279 OR Insertion |
ユーザー |
![]() |
提出日時 | 2023-04-21 22:53:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 406 ms / 2,000 ms |
コード長 | 606 bytes |
コンパイル時間 | 312 ms |
コンパイル使用メモリ | 82,480 KB |
実行使用メモリ | 78,012 KB |
最終ジャッジ日時 | 2024-11-06 16:20:58 |
合計ジャッジ時間 | 12,330 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 48 |
ソースコード
import sys input = sys.stdin.readline mod = 998244353 N = int(input()) S = list(map(int, input().rstrip())) pow2 = [1]*N for i in range(1, N): pow2[i] = 2*pow2[i-1] % mod ans = 0 for k in range(N): dp = [0] * (N+1) dp[0] = 1 cumsum = [0] * (N+2) cnt = 0 for i in range(1, N+1): cumsum[i] = (cumsum[i-1] + dp[i-1]) % mod if i-k-1 >= 0 and S[i-k-1] == 1: dp[i] = (cumsum[i] - cumsum[i-k]) % mod cnt = (cnt + cumsum[i-k] * pow2[max(0, N-i-1)]) % mod else: dp[i] = cumsum[i] ans = (ans + cnt * pow2[k]) % mod print(ans)