結果

問題 No.2019 Digits Filling for All Substrings
ユーザー Shirotsume
提出日時 2022-02-13 11:01:43
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,553 ms / 2,000 ms
コード長 835 bytes
コンパイル時間 133 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 28,160 KB
最終ジャッジ日時 2024-06-29 05:29:51
合計ジャッジ時間 21,425 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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