백준 1920번. 수 찾기

https://www.acmicpc.net/problem/1920

1920번: 수 찾기

import sys
input = sys.stdin.readline

n = int(input())
a = set(map(int, input().split())) #set을 통해 시간절약

m = int(input())
b = list(map(int, input().split()))

for i in b:
  if str(i) in str(a):
    print(1)
  else :
    print(0)

시간초과가 뜬 코드이다.

이분탐색으로 해결해야하나 싶었지만,

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)

뭔 차이지????? 통과가 되었다...

 

import sys
input = sys.stdin.readline

n = int(input())
a = list(map(int, input().split()))

m = int(input())
b = list(map(int, input().split()))

a.sort()


for i in b:
  lt, rt = 0, n-1
  isin = False # 안에 있는지

  #이분탐색 시작
  while lt <= rt : #lt가 rt보다 커지면 탈출
    mid = (lt + rt) // 2
    if i == a[mid]:
      isin = True
      print(1)
      break
    elif i > a[mid]: #왼쪽 값이 더 크면
      lt = mid + 1
    else : rt = mid -1

  if not isin:
    print(0)

이분탐색 코드는 직접 다시 짜봐야할 듯