結果

問題 No.2019 Digits Filling for All Substrings
ユーザー Shirotsume
提出日時 2022-02-07 11:02:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 301 ms / 2,000 ms
コード長 835 bytes
コンパイル時間 286 ms
コンパイル使用メモリ 81,912 KB
実行使用メモリ 81,128 KB
最終ジャッジ日時 2024-06-12 01:40:31
合計ジャッジ時間 6,353 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def solve(n,s):
    mod = 998244353
    a = [0] * (n + 1)
    for i in range(n):
        if s[i] == '?':
            a[i + 1] += 1
        a[i + 1] += a[i]
    A = [0] * (n + 1)

    for i in range(n + 1):
        A[i] = pow(10, a[i], mod)
    ans = 0
    now = 0
    for i in range(n + 1):
        ans += A[i] * now
        now += pow(A[i], mod - 2, mod)
    ans = (ans - (n + 1) * (n) * pow(2, mod - 2, mod)) * pow(3, mod - 2, mod)
    a = [0] * (n + 1)
    b = [0] * 3
    b[0] = 0
    for i in range(n):
        if s[i] == '?':
            a[i + 1] += a[i]
            continue
        else: 
            a[i + 1] += int(s[i]) + a[i]
    for i in a:
        b[i % 3] += 1
    for i in range(3):
        ans += b[i] * (b[i] - 1) // 2
        ans %= mod
    ans %= mod
    return ans

n = int(input())
s = input()

print(solve(n, s))
0