結果

問題 No.662 スロットマシーン
コンテスト
ユーザー flippergo
提出日時 2026-03-09 10:12:50
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 357 ms / 2,000 ms
コード長 1,237 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 342 ms
コンパイル使用メモリ 85,572 KB
実行使用メモリ 84,900 KB
最終ジャッジ日時 2026-03-09 10:32:01
合計ジャッジ時間 5,406 ms
ジャッジサーバーID
(参考情報)
judge1_1 / judge2_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

pict = {}
ind2s = {}
s2ind = {}
for i in range(5):
    s,n = input().split()
    ind2s[i] = s
    s2ind[s] = i
    pict[s] = int(n)
col1 = {}
n1 = int(input())
A = [input().strip() for _ in range(n1)]
A = A+A
for i in range(n1):
    x = tuple(A[i:i+3])
    col1[x] = col1.get(x,0)+1
col2 = {}
n2 = int(input())
B = [input().strip() for _ in range(n2)]
B = B+B
for i in range(n2):
    x = tuple(B[i:i+3])
    col2[x] = col2.get(x,0)+1
col3 = {}
n3 = int(input())
C = [input().strip() for _ in range(n3)]
C = C+C
for i in range(n3):
    x = tuple(C[i:i+3])
    col3[x] = col3.get(x,0)+1
ans = [0]*5
prob = 0
for a in col1:
    for b in col2:
        for c in col3:
            for i in range(3):
                if a[i]==b[i]==c[i]:
                    ans[s2ind[a[i]]] += col1[a]*col2[b]*col3[c]
                    prob += col1[a]*col2[b]*col3[c]/n1/n2/n3*pict[a[i]]
            if a[0]==b[1]==c[2]:
                ans[s2ind[a[0]]] += col1[a]*col2[b]*col3[c]
                prob += col1[a]*col2[b]*col3[c]/n1/n2/n3*pict[a[0]]
            if a[2]==b[1]==c[0]:
                ans[s2ind[a[2]]] += col1[a]*col2[b]*col3[c]
                prob += col1[a]*col2[b]*col3[c]/n1/n2/n3*pict[a[2]]
print(prob)
for i in range(5):
    print(ans[i])
0