結果
問題 | No.2964 Obstruction Bingo |
ユーザー | sasa8uyauya |
提出日時 | 2024-11-16 16:26:58 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 778 bytes |
コンパイル時間 | 526 ms |
コンパイル使用メモリ | 82,016 KB |
実行使用メモリ | 193,532 KB |
最終ジャッジ日時 | 2024-11-16 16:29:29 |
合計ジャッジ時間 | 147,889 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 TLE * 41 |
ソースコード
L,K=map(int,input().split()) s=list(input()) for i in range(L): s[i]=ord(s[i])-ord("a") t=list(input()) for i in range(L): t[i]=ord(t[i])-ord("a") a=list(map(int,input().split())) M=998244353 A=sum(a) iA=pow(A,M-2,M) X=10**10 w1=0 w2=0 q=[[[0]*(2*L+1) for j in range(L)] for i in range(L)] q[0][0][0]=1 for _ in range(K): nq=[[[0]*(2*L+1) for j in range(L)] for i in range(L)] for i in range(L): for j in range(L): for l in range(-L+1,L): for c in range(26): nq[(i+(c==s[i]))%L][(j+(c==t[j]))%L][l+(c==s[i])-(c==t[j])]+=q[i][j][l]*a[c]*iA nq[(i+(c==s[i]))%L][(j+(c==t[j]))%L][l+(c==s[i])-(c==t[j])]%=M q=nq for i in range(L): for j in range(L): w1+=q[i][j][L] w1%=M w2+=q[i][j][-L] w2%=M print(w1,w2)