CS/알고리즘_문제풀이(파이썬)

숫자박스

Jedy_Kim 2021. 6. 30. 14:03
728x90

문제

숫자박스에는 자연수들이 적혀있는 종이들이 N장 들어있다. 숫자 M개가 주어졌을 때, 이 숫자가 적혀있는 숫자가 상자 안에 있는지 아닌지를 구하는 프로그램을 작성하시오.  

입력

첫째 줄에 윤성이가 가지고 있는 숫자 정보의 개수 N (1 ≤ N ≤ 300,000)이가 주어진다. 둘째 줄에는 숫자 정보들이 주어진다. 숫자는 절댓값 1,000,000 이하의 정수이다. 두 숫자 카드에 같은 숫자가 적혀있는 경우는 없다.

셋째 줄에는 M (1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 윤성이가 숫자 박스에 있는지 아닌지를 구해야 할 M개의 숫자가 주어지며, 이 숫자는 공백으로 구분되어져 있다. 이숫자도 1,000,000이하의 정수이다.  

출력

첫째 줄에 입력으로 주어진 M개의 숫자에 대해서, 각 숫자가 숫자 상자안에 있으면 1, 아니면 0을 개행으로 구분해 출력한다.

 

예제 입력

5

6 3 2 10 -10

8

10 9 -5 2 3 4 5 -10

예제 출력

1

0

0

1

1

0

0

1

 

#코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from bisect import bisect_left, bisect_right
import sys
 
if __name__ == "__main__":
  input = sys.stdin.readline
  n        = int(input())
  origin   = list(map(int, input().split())) 
  m        = int(input())
  myArr    = list(map(int, input().split()))
  len_list = len(origin)
  
  origin.sort()
  for i in myArr:
    result_left = bisect_left(origin, i)
    
    if result_left < len_list:
      if origin[result_left] == i:
        print(1)
      else:
        print(0)
    else:
      print(0)
cs

 

반응형

'CS > 알고리즘_문제풀이(파이썬)' 카테고리의 다른 글

2차식 정답 추측  (0) 2021.07.02
두 용액  (0) 2021.06.30
이진탐색  (0) 2021.06.30
퀵정렬 구현하기  (0) 2021.06.30
N-Queen  (0) 2021.06.29