1920번: 수 찾기

1920번: 수 찾기

import sys
input = sys.stdin.readline
N = int(input())
A = set(map(int,input().split()))
M = int(input())
B = list(map(int,input().split()))

for i in B:
    print(1) if i in A else print(0)

원하는 값 출력은 나왔으나 계속 런타임에러가 나와서 이것저것 수정해보았다.
시간을 줄이기 위해 변경한 방법 3가지는
1. input 대신 입력값이 받는 시간이 더 짧은 sys.stdin.readline 을 사용하였다.
2. A 자료형을 리스트가 아닌 set 자료형으로 받았다.
3. for-if else문을 한 줄로 작성하였다.

이 세가지를 모두 고치니 그제서야 성공이 나왔다.
1번과 3번은 import 함수를 사용하든 for-if else문을 3줄로 작성을 하든 성공여부가 큰 차이는 없는 것 같다.
하지만 2번 set 자료형을 list로 바꾸게되면 시간초가가 발생한다.
즉 2번의 내용이 시간 초과 여부의 키포인트가 되는 것 같다.