[이것이 코딩테스트다] Chap07. 이진탐색-연습문제

SunYerim·2023년 1월 19일
0
post-thumbnail

1. 부품찾기

동빈이네 전자 매장에는 부품이 N개 있다. 각 부품은 정수 형태의 고유한 번호가 있다. 어느 날 손님이 M개의 종류의 부품을 대량으로 구매하겠다며 당일 날 견적서를 요청했다. 동빈이는 때를 놓치지 않고 손님이 문의한 부품 M개의 종류를 모두 확인해서 견적서를 작성해야 한다. 이때 가게 안에 부품이 모두 있는지 확인하는 프로그램을 작성해보자.
이때, 손님이 요청한 부품 번호의 순서대로 부품을 확인해 부품이 있으면 yes, 없으면 no를 출력한다.

입력조건

첫째 줄에 정수 N이 주어진다. (1 <= N <= 1,000,000)
둘째 줄에는 공백으로 구분하여 N개의 정수가 주어진다. 이때 정수는 1보다 크고, 1,000,000 이하이다.
셋째 줄에는 정수 M이 주어진다. (1 <= M <= 100,000)
넷째 줄에는 공백으로 구분하여 M개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.

출력조건

첫째 줄에 공백으로 구분하여 각 부품이 존재하면 yes를, 없으면 no를 출력한다.

입력예시

5
8 3 7 9 2
3
5 7 9

출력예시

no yes yes

내 logic과 풀이 코드

'''logic
    1. 필요한 내용 모두 입력받은 후
    2. 재귀로 이진탐색 실행'''

def binary_search(array, target, start, end):
    # 아예 조건 성립이 안 되는 경우 배제
    while start <= end:
        mid = (start + end) // 2
        # 찾고자 하는 target이 중간값인 경우
        if (array[mid] == target):
            return mid
        # 찾고자하는 target이 중간값의 값보다 작은 경우
        elif (array[mid] > target):
            end = mid - 1
        # 찾고자하는 target이 중간값의 값보다 큰 경우
        else:
            start = mid + 1
    return None
    
N = int(input())
market_array = list(map(int, input().split()))
market_array.sort() # 이진 탐색 위해 정렬

M = int(input())
consumer_array = list(map(int, input().split()))

for i in consumer_array:
    # 해당 부품 존재하는지 binary search 통해 확인
    result = binary_search(market_array, i, 0, N-1)
    if (result != None):
        print('yes', end = ' ')
    else:
        print('no', end = ' ')

2. 떡볶이 떡 만들기

오늘 동빈이는 여행 가신 부모님을 대신해서 떡집 일을 하기로 했다. 오늘은 떡볶이 떡을 만드는 날이다. 동빈이네 떡볶이 떡은 재밌게도 떡볶이의 떡의 길이가 일정하지 않다. 대신에 한 봉지 안에 들어가는 떡의 총 길이는 절단기로 잘라서 맞춰준다.
절단기에 높이 (H)를 지정하면 줄지어진 떡을 한 번에 절단한다. 높이가 H보다 긴 떡은 H 위의 부분이 잘릴 것이고, 낮은 떡은 잘리지 않는다.
예를 들어 높이가 19, 14, 10, 17cm인 떡이 나란히 있고 절단기 높이를 15cm으로 지정하면 자른 뒤 떡의 높이는 15, 14, 10 15cm이 될 것이다. 잘린 떡의 길이는 차례대로 4, 0, 0, 2cm이다. 손님은 6cm만큼의 길이를 가져간다.
손님이 왔을 때 요청한 총 길이가 M일 때 적어도 M만큼의 떡을 얻기 위해 절단기에 설정할 수 있는 높이의 최댓값을 구하는 프로그램을 작성하시오.

입력조건

첫째 줄에 떡의 개수 N과 요청한 떡의 길이 M이 주어진다. (1 <= N <= 1,000,000 / 1 <= M <= 2,000,000,000)
둘째 줄에는 떡의 개별 높이가 주어진다. 떡 높이의 총합은 항상 M 이상이므로, 손님은 필요한 양만큼 떡을 사갈 수 있다. 높이는 10억보다 작거나 같은 양의 정수 또는 0이다.

출력조건

적어도 M만큼의 떡을 집에 가져가기 위해 절단기에 설정할 수 있는 높이의 최댓값을 출력한다.

입력예시

4 6
19 15 10 17

출력예시

15

내 logic과 풀이 코드

N, M = map(int, input().split())

rice = list(map(int, input().split()))
rice.sort()

# 시작점과 끝점 설정
start = 0
end = rice[-1]

# 정답 변수
result = 0

# 이진 탐색으로
while (start <= end):
    total = 0
    mid = (start + end) // 2
    for x in rice:
        # 잘랐을때 떡의 양 계산
        if (x > mid):
            total += x - mid
       
    if (total < M):
        end = mid - 1
    else:
        result = mid
        start = mid + 1

print(result)
profile
내 안에 있는 힘을 믿어라.

0개의 댓글