結果

問題 No.2019 Digits Filling for All Substrings
コンテスト
ユーザー asumo0729
提出日時 2022-08-09 22:46:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 425 ms / 2,000 ms
コード長 764 bytes
コンパイル時間 340 ms
コンパイル使用メモリ 81,908 KB
実行使用メモリ 92,460 KB
最終ジャッジ日時 2024-09-20 03:38:02
合計ジャッジ時間 8,176 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
N = int(input())
S = input()
acc = [0]
for s in S:
    if s == '?':
        acc.append(acc[-1] + 1)
    else:
        acc.append(acc[-1])
    acc[-1] %= mod
inv3 = pow(3, mod - 2, mod)
inv_10 = [1]
for i in range(N):
    x = pow(10, acc[i + 1], mod)
    inv_10.append(pow(x, mod - 2, mod))
for i in range(N):
    inv_10[i + 1] += inv_10[i]
    inv_10[i + 1] %= mod
ans = 0
ans -= (N + 1) * N // 2
ans %= mod
for r in range(1, N + 1):
    ans += pow(10, acc[r], mod) * inv_10[r - 1]
    ans %= mod
ans *= inv3
ans %= mod

mod3 = [0 for _ in range(3)]
mod3[0] += 1
res = 0
for s in S:
    if s == '?':
        x = 0
    else:
        x = int(s)
    res += x
    mod3[res % 3] += 1
for x in mod3:
    ans += x * (x - 1) // 2
    ans %= mod
print(ans)
0