結果
問題 | No.1845 Long Substrings |
ユーザー | ygd. |
提出日時 | 2022-02-18 22:35:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 140 ms / 2,000 ms |
コード長 | 1,211 bytes |
コンパイル時間 | 278 ms |
コンパイル使用メモリ | 82,056 KB |
実行使用メモリ | 104,924 KB |
最終ジャッジ日時 | 2024-06-29 09:21:54 |
合計ジャッジ時間 | 4,175 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 34 |
ソースコード
import sys #input = sys.stdin.readline #文字列につけてはダメ #input = sys.stdin.buffer.readline #文字列につけてはダメ #sys.setrecursionlimit(1000000) #import bisect #import itertools #import random #from heapq import heapify, heappop, heappush #from collections import defaultdict #from collections import deque import copy #import math #from functools import lru_cache #@lru_cache(maxsize=None) MOD = pow(10,9) + 7 #MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] def main(): N = int(input()) A = list(map(int,input().split())) S = str(input()) #dp[i][j]:i番目Aまで見て、アルファベットjが最後の時の場合の数。 #j == 0は空文字列 dp = [0]*27 #dp[0] = 1 for i in range(N): p = copy.copy(dp) #追加しないとき dp,p = p,dp v = A[i] s = ord(S[i]) - ord('a') + 1 for j in range(27): #追加するとき if j == s: #同じ時はv個だけ最大が伸びる dp[s] += v else: dp[s] += p[j] * v dp[s] %= MOD #print(dp) ans = sum(dp)%MOD print(ans) if __name__ == '__main__': main()