[python] 부품 찾기

전.py / / 2022. 7. 14. 18:14
728x90

- 손님이 요청한 부품이 있는지 확인

# 이진 탐색
def binary_search(array, target, start, end):
    while start <= end:
        mid = (start + end) // 2
        
        if array[mid] == target:
            return mid
        
        elif array[mid] > target:
            end = mid - 1
            
        else:
            start = mid + 1
    return None


n = int(input())
data = list(map(int, input().split()))
data.sort()

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

for i in inp:
    result = binary_search(data, i, 0, n-1)
    if result != None:
        print('yes', end=' ')
    else:
        print('no', end=' ')
# 계수 정렬
n = int(input())
data = [0] * 1000001

for i in input().split():
    data[int(i)] = 1
    
m = int(input())
inp = list(map(int, input().split()))

for i in inp:
    if data[i] == 1:
        print('yes', end=' ')
    else:
        print('no', end=' ')
# set 자료형 이용
n = int(input())
data = set(map(int, input().split()))

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

for i in inp:
    if i in data:
        print('yes', end=' ')
    else:
        print('no', end=' ')

반응형

'전.py' 카테고리의 다른 글

[python] 1로 만들기  (0) 2022.07.21
[python] 떡볶이 떡 만들기  (0) 2022.07.21
[python] 이진 탐색  (0) 2022.07.14
[python] 두 배열의 원소 교체  (0) 2022.07.12
[python] 성적이 낮은 순서로 학생 출력하기  (0) 2022.07.12
  • 네이버 블러그 공유하기
  • 네이버 밴드에 공유하기
  • 페이스북 공유하기
  • 카카오스토리 공유하기