結果

問題 No.1156 Nada Picnic 2
ユーザー lloyzlloyz
提出日時 2022-07-08 15:28:24
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 748 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 82,324 KB
実行使用メモリ 76,468 KB
最終ジャッジ日時 2024-06-08 09:45:53
合計ジャッジ時間 3,578 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 106 ms
75,776 KB
testcase_01 AC 611 ms
75,776 KB
testcase_02 TLE -
権限があれば一括ダウンロードができます

ソースコード

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

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]]] * pow(10, len(s) - d - 1)
    if ANS[0] + ANS[1] == ANS[2]:
        print(ANS[2])
        exit()
0