[백준/파이썬]13118번 뉴턴과 사과

-코드

p = list(map(int, input().split()))
x, y, r = map(int, input().split())
r = 0
for i in range(4):
    if x == p[i]:
        r = i + 1
        break
print(r)