[Python] ABC003 B

B – AtCoderトランプ

s = input()
t = input()

commercial_at = ['a', 't', 'c', 'o', 'd', 'e','r']

flag = True
for i in range(len(s)):
    if s[i] == t[i]:
        continue
    elif s[i] == '@':
        if t[i] in commercial_at:
            continue
        else:
            flag = False
            break
    elif t[i] == '@':
        if s[i] in commercial_at:
            continue
        else:
            flag = False
            break
    else:
        flag = False
        break

if flag:
    print('You can win')
else:
    print('You will lose')

メモ

• 勝ち、負けのフラグを作る。
• 一文字ずつ判定する。