結果

問題 No.1845 Long Substrings
ユーザー ああいいああいい
提出日時 2022-02-18 23:02:11
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 104 ms / 2,000 ms
コード長 628 bytes
コンパイル時間 133 ms
コンパイル使用メモリ 82,456 KB
実行使用メモリ 108,140 KB
最終ジャッジ日時 2024-06-29 09:43:09
合計ジャッジ時間 3,428 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = list(map(int,input().split()))
S = input()
last = [-1] * 26
P = 10 ** 9 + 7
dp = [0] * N
_sum = [0] * (N+1)
dp[0] = A[0]
c = ord(S[0]) - ord('a')
last[c] = 0
_sum[1] = A[0]
hoka = [0] * N
hoka[0] = 1
for i in range(1,N):
    c = ord(S[i]) - ord('a')
    if last[c] == -1:
        dp[i] = (_sum[i]+1) * A[i] % P
        hoka[i] = _sum[i]+1
    else:
        dp[i] = (_sum[i] - _sum[last[c]+1] + hoka[last[c]] ) * A[i] % P
        hoka[i] = (_sum[i] - _sum[last[c]+1] + hoka[last[c]]) % P
    _sum[i+1] = (_sum[i] + dp[i]) % P
    last[c] = i
ans = 0
for i in range(N):
    ans = (ans + dp[i]) % P
print(ans)
0