結果

問題 No.2019 Digits Filling for All Substrings
ユーザー ShirotsumeShirotsume
提出日時 2022-02-03 22:33:10
言語 PyPy3
(7.3.15)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 805 bytes
コンパイル時間 231 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 81,268 KB
最終ジャッジ日時 2024-06-12 01:41:32
合計ジャッジ時間 5,571 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 7 WA * 23
権限があれば一括ダウンロードができます

ソースコード

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] = 1
    for i in range(n):
        if s[i] == '?':
            b[0] += 1
            continue
        else: a[i + 1] += int(s[i])
        a[i + 1] += a[i]
        b[a[i + 1] % 3] += 1
    for i in range(3):
        ans += b[i] * (b[i] - 1) // 2
    ans %= mod
    return ans

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

print(solve(n, s))
0