結果

問題 No.1845 Long Substrings
ユーザー lilictaka
提出日時 2022-05-09 17:42:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,086 ms / 2,000 ms
コード長 775 bytes
コンパイル時間 250 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 129,772 KB
最終ジャッジ日時 2024-07-16 15:48:24
合計ジャッジ時間 15,431 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_left
def ch(s):
    return ord(s) - ord('a')
N = int(input())
memo = [[] for _ in range(26)]
A = list(map(int,input().split()))
mod = 10 ** 9 + 7
S = list(map(ch,list(input())))
dp = [0 for _ in range(N)]
for i in range(N):
    if memo[S[i]] == []:
        dp[i] = 1
    memo[S[i]].append(i)
for i in range(N):
    for k in range(26):
        if memo[k] == []:
            pass
        else:
            index = bisect_left(memo[k],i+1)
            if index == len(memo[k]):
                continue
            nx = memo[k][index]
            if S[i] == k:
                dp[nx] += dp[i]
            else:
                dp[nx] += dp[i] * A[i]
            dp[nx] %= mod
ans = 0
for i in range(N):
    ans += dp[i] * A[i]
    ans %= mod
print(ans)
0