結果
| 問題 |
No.2019 Digits Filling for All Substrings
|
| コンテスト | |
| ユーザー |
👑 SPD_9X2
|
| 提出日時 | 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)
SPD_9X2