10815번: 숫자 카드
문제 )
숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.
입력 :
첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.
셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다
출력 :
첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.
풀이)
이진탐색을 이용하여 풀어보았다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
n = int(input())
li = sorted(list(map(int, input().split())))
m = int(input())
arr = list(map(int, input().split()))
for i in arr:
s = 0
e = n - 1
while s <= e:
mid = (s + e) // 2
if li[mid] == i:
print(1, end=" ")
break
elif li[mid] < i:
s = mid + 1
else:
e = mid - 1
else:
print(0, end=" ")
|
cs |
'Develop > 백준 (python)' 카테고리의 다른 글
[백준] 2792번: 보석 상자 (python) (0) | 2022.03.25 |
---|---|
[백준] 18113번: 그르다 김가놈 (python) (0) | 2022.03.25 |
[백준] 15565번: 귀여운 라이언 (python) (0) | 2022.03.24 |
[백준] 14846번: 직사각형과 쿼리 (python) (0) | 2022.03.23 |
[백준] 11660번: 구간 합 구하기 5 (Silver) (python) (0) | 2022.03.23 |