結果
問題 | No.2019 Digits Filling for All Substrings |
ユーザー |
![]() |
提出日時 | 2022-07-22 22:08:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 118 ms / 2,000 ms |
コード長 | 912 bytes |
コンパイル時間 | 526 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 85,760 KB |
最終ジャッジ日時 | 2024-07-04 06:31:40 |
合計ジャッジ時間 | 3,630 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
""" 2019: ?がいくつ含まれるか 他の桁の総和mod3はいくつか これが列挙できればok ?の個数が 1~|S| 他の桁mod3が、0~2 なので、列挙さえできれば O(|S|) 累積和的に考える? 右端からの累積和を考えよう 0の個数が、そこ右端での答え dpで行けるかな? """ import sys from sys import stdin N = int(stdin.readline()) S = list(stdin.readline()[:-1]) mod = 998244353 ans = 0 dp = [0,0,0] for i in range(N): ndp = [0,0,0] if S[i] == "?": for nc in range(10): for k in range(3): ndp[ (k+nc) % 3 ] += dp[k] ndp[nc % 3] += 1 else: nc = int(S[i]) for k in range(3): ndp[ (k+nc) % 3 ] += dp[k] ndp[nc % 3] += 1 ans += ndp[0] ans %= mod for j in range(3): ndp[j] %= mod dp = ndp print (ans % mod)