結果

問題 No.2964 Obstruction Bingo
ユーザー ねしんねしん
提出日時 2024-10-24 17:57:57
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 868 ms / 2,468 ms
コード長 1,550 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 82,448 KB
実行使用メモリ 98,616 KB
最終ジャッジ日時 2024-10-26 23:58:04
合計ジャッジ時間 24,935 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,568 KB
testcase_01 AC 37 ms
53,020 KB
testcase_02 AC 47 ms
62,332 KB
testcase_03 AC 36 ms
53,192 KB
testcase_04 AC 37 ms
53,648 KB
testcase_05 AC 62 ms
74,140 KB
testcase_06 AC 414 ms
86,340 KB
testcase_07 AC 95 ms
77,236 KB
testcase_08 AC 308 ms
83,496 KB
testcase_09 AC 87 ms
77,320 KB
testcase_10 AC 235 ms
81,600 KB
testcase_11 AC 282 ms
83,840 KB
testcase_12 AC 616 ms
92,128 KB
testcase_13 AC 138 ms
78,636 KB
testcase_14 AC 128 ms
78,336 KB
testcase_15 AC 75 ms
76,404 KB
testcase_16 AC 213 ms
81,328 KB
testcase_17 AC 168 ms
79,616 KB
testcase_18 AC 166 ms
79,240 KB
testcase_19 AC 266 ms
83,020 KB
testcase_20 AC 348 ms
84,728 KB
testcase_21 AC 596 ms
92,116 KB
testcase_22 AC 200 ms
81,544 KB
testcase_23 AC 117 ms
78,632 KB
testcase_24 AC 55 ms
68,468 KB
testcase_25 AC 826 ms
98,264 KB
testcase_26 AC 830 ms
98,088 KB
testcase_27 AC 826 ms
98,040 KB
testcase_28 AC 824 ms
98,288 KB
testcase_29 AC 829 ms
98,256 KB
testcase_30 AC 821 ms
97,892 KB
testcase_31 AC 827 ms
97,940 KB
testcase_32 AC 834 ms
98,080 KB
testcase_33 AC 831 ms
98,260 KB
testcase_34 AC 836 ms
97,916 KB
testcase_35 AC 837 ms
98,088 KB
testcase_36 AC 850 ms
98,204 KB
testcase_37 AC 854 ms
98,616 KB
testcase_38 AC 833 ms
98,148 KB
testcase_39 AC 856 ms
98,136 KB
testcase_40 AC 837 ms
97,944 KB
testcase_41 AC 831 ms
98,248 KB
testcase_42 AC 868 ms
98,240 KB
testcase_43 AC 844 ms
98,192 KB
testcase_44 AC 864 ms
98,240 KB
testcase_45 AC 121 ms
78,084 KB
testcase_46 AC 41 ms
54,352 KB
testcase_47 AC 40 ms
55,236 KB
testcase_48 AC 57 ms
67,664 KB
testcase_49 AC 128 ms
78,328 KB
testcase_50 AC 298 ms
83,508 KB
testcase_51 AC 754 ms
96,284 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD=998244353
L,K=list(map(int,input().split()))
S=input()
T=input()
A=list(map(int,input().split()))
SUM=sum(A)
INVSUM=pow(SUM,MOD-2,MOD)
s=[]
t=[]
for i in range(L):
  s.append(ord(S[i])-ord("a"))
  t.append(ord(T[i])-ord("a"))
#print(s,t)  
dp=[]
for i in range(K+1):
  d=[]
  for j in range(L):
    d.append([0]*(2*L+1))
  dp.append(d)
dp[0][0][L]=1#dp[何回目の試行か][ナナの次の文字][ミンサ-ナナ,ポイント差+L]
for i in range(K):
  for j in range(L):
    for k in range(1,2*L):
      if dp[i][j][k]==0:
        continue
      nanac=j
      minsac=(j+k)%L
      if s[nanac]==t[minsac]:
        dp[i+1][nanac][k]=(dp[i+1][nanac][k]+dp[i][nanac][k]*(SUM-A[s[nanac]])*INVSUM)%MOD#違う文字が出た場合
        dp[i+1][(nanac+1)%L][k]=(dp[i+1][(nanac+1)%L][k]+dp[i][nanac][k]*A[s[nanac]]*INVSUM)%MOD#二人とも1ポイント獲得
      else:
        dp[i+1][nanac][k]=(dp[i+1][nanac][k]+dp[i][nanac][k]*(SUM-A[s[nanac]]-A[t[minsac]])*INVSUM)%MOD#違う文字が当選
        dp[i+1][(nanac+1)%L][k-1]=(dp[i+1][(nanac+1)%L][k-1]+dp[i][nanac][k]*A[s[nanac]]*INVSUM)%MOD#ナナが当選
        dp[i+1][nanac][k+1]=(dp[i+1][nanac][k+1]+dp[i][nanac][k]*A[t[minsac]]*INVSUM)%MOD#ミンサが当選
        #print(dp[i+1][nanac][k],dp[i+1][(nanac+1)%L][k-1],dp[i+1][nanac][k+1],dp[i][nanac][k])
    dp[i+1][j][0]=(dp[i+1][j][0]+dp[i][j][0])%MOD
    dp[i+1][j][2*L]=(dp[i+1][j][2*L]+dp[i][j][2*L])%MOD

ansN=0
ansM=0
#print(dp)
for i in range(L):
  ansN=(ansN+dp[K][i][0])%MOD
  ansM=(ansM+dp[K][i][2*L])%MOD
print(ansN,ansM)
0