알고리즘/파이썬

[파이썬 알고리즘] 백준 수 찾기

귤먹는코더 2022. 5. 19. 01:46
728x90

 

풀이) 풀이를 보면 굉장히 쉬운 문제다. 정답률이 20%대라는게 믿기지 않는 문제고,

나는 기존에 input() 함수만 입력값을 넣는 줄 알았는데, 찾아보니까 sys.stdin.readline 이라는 게 있었다.

from sys import stdin, stdout
n = stdin.readline()
N = stdin.readline().split()
m = stdin.readline()
M = stdin.readline().split()

for l in M:
    stdout.write("0\n") if l not in N else stdout.write('1\n')

input 대신 sys.stdin.readline 쓰는 이유

- 여러 줄을 입력 받아야 할 때는 input()으로 입력 받는다면 시간초과가 발생할 수 있다.

 

stdout.write("0\n") if l not in N else stdout.write('1\n')

만약 N안에 l이 없다면 0을 적고, 아니면 1을 적어라를 이렇게 적을 수 도 있다는 것을 알았다.

728x90