結果
問題 | No.1845 Long Substrings |
ユーザー |
![]() |
提出日時 | 2022-02-18 22:13:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 447 ms / 2,000 ms |
コード長 | 1,520 bytes |
コンパイル時間 | 192 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 171,552 KB |
最終ジャッジ日時 | 2024-06-29 09:05:16 |
合計ジャッジ時間 | 8,445 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 34 |
ソースコード
""" 部分列dpを思い出せ… dp[i][flag] = i番目の区間の文字を1つ以上使用しており、 最後の文字を使っているかどうかのflagが flagの時の場合の数 とすれば解ける flagが Falseなら同じ文字の区間には飛べない """ import sys from sys import stdin mod = 10**9+7 N = int(stdin.readline()) A = list(map(int,stdin.readline().split())) S = list(stdin.readline()[:-1]) A.append(1) S.append("{") lastinds = [None] * 27 indlis = [] for i in range(N,-1,-1): indlis.append( tuple(lastinds) ) lastinds[ ord(S[i]) - ord("a") ] = i indlis.reverse() #print (indlis) dp = [[0,0] for i in range(N+1)] startend = [None] * 27 for i in range(N): #自分スタートの処理 if startend[ord(S[i]) - ord("a")] == None: startend[ord(S[i]) - ord("a")] = 1 dp[i][0] += A[i]-1 dp[i][1] += 1 #推移 for k in range(27): nexind = indlis[i][k] if nexind == None: continue if k != ord(S[i])-ord('a'): dp[nexind][0] += dp[i][0] * (A[nexind]-1) dp[nexind][1] += dp[i][0] dp[nexind][0] += dp[i][1] * (A[nexind]-1) dp[nexind][1] += dp[i][1] else: #dp[nexind][0] += dp[i][0] * (A[nexind]-1) #dp[nexind][1] += dp[i][0] dp[nexind][0] += dp[i][1] * (A[nexind]-1) dp[nexind][1] += dp[i][1] dp[nexind][0] %= mod dp[nexind][1] %= mod #print (dp) print (dp[-1][1] % mod)