結果

問題 No.1156 Nada Picnic 2
ユーザー lloyz
提出日時 2024-06-12 21:01:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,795 ms / 2,000 ms
コード長 786 bytes
コンパイル時間 556 ms
コンパイル使用メモリ 82,452 KB
実行使用メモリ 76,368 KB
最終ジャッジ日時 2024-06-12 21:01:16
合計ジャッジ時間 3,782 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import permutations

n = int(input())

if n == 1:
    a = 'abc'
    b = 'def'
    c = 'bgcb'
elif n == 2:
    a = 'aabc'
    b = 'defg'
    c = 'hibcj'
elif n == 3:
    a = 'spring'
    b = 'eight'
    c = 'picnic'

S = set()
for s in [a, b, c]:
    for ss in s:
        S.add(ss)
num = len(S)
L = list(S)
s_to_i = dict()
for i in range(num):
    s_to_i[L[i]] = i
Pow10 = [pow(10, i) for i in range(11)]

for P in permutations(range(10), num):
    if P[s_to_i[a[0]]] == 0 or P[s_to_i[b[0]]] == 0 or P[s_to_i[c[0]]] == 0:
        continue
    ANS = [0, 0, 0]
    for i in range(3):
        s = [a, b, c][i]
        for d in range(len(s)):
            ANS[i] += P[s_to_i[s[d]]] * Pow10[len(s) - d - 1]
    if ANS[0] + ANS[1] == ANS[2]:
        print(ANS[2])
        exit()
0