結果
問題 | No.2541 Divide 01 String |
ユーザー |
|
提出日時 | 2023-11-24 23:12:47 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 73 ms / 2,000 ms |
コード長 | 356 bytes |
コンパイル時間 | 185 ms |
コンパイル使用メモリ | 81,952 KB |
実行使用メモリ | 76,672 KB |
最終ジャッジ日時 | 2024-09-26 09:45:49 |
合計ジャッジ時間 | 2,325 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
N = int(input()) S = input() p = 998244353 dp = [0] * (N + 1) dp[0] = 1 cum = [0] * (N + 1) cum[1] = 1 j = -1 for i, s in enumerate(S): if s == "1": j = i if j == -1: cum[i + 1] += cum[i] cum[i + 1] %= p continue cum[i + 1] = (cum[i] + dp[i]) % p dp[i + 1] = cum[j + 1] print(dp[N]) # print(dp) # print(cum)