[Silver III] 이상한 술집 - 13702

문제 링크

성능 요약

메모리: 31256 KB, 시간: 72 ms

분류

이분 탐색

문제 설명

프로그래밍 대회 전날, 은상과 친구들은 이상한 술집에 모였다. 이 술집에서 막걸리를 시키면 주전자의 용량은 똑같았으나 안에 들어 있는 막걸리 용량은 랜덤이다. 즉 한 번 주문에 막걸리 용량이 802ml 이기도 1002ml가 나오기도 한다. 은상은 막걸리 N 주전자를 주문하고, 자신을 포함한 친구들 K명에게 막걸리를 똑같은 양으로 나눠주려고 한다. 그런데 은상과 친구들은 다른 주전자의 막걸리가 섞이는 것이 싫어서, 분배 후 주전자에 막걸리가 조금 남아 있다면 그냥 막걸리를 버리기로 한다. (즉, 한 번 주문한 막걸리에 남은 것을 모아서 친구들에게 다시 주는 경우는 없다. 예를 들어 5명이 3 주전자를 주문하여 1002, 802, 705 ml의 막걸리가 각 주전자에 담겨져 나왔고, 이것을 401ml로 동등하게 나눴을 경우 각각 주전자에서 200ml, 0m, 304ml 만큼은 버린다.) 이럴 때 K명에게 최대한의 많은 양의 막걸리를 분배할 수 있는 용량 ml는 무엇인지 출력해주세요.

입력

첫째 줄에는 은상이가 주문한 막걸리 주전자의 개수 N, 그리고 은상이를 포함한 친구들의 수 K가 주어진다. 둘째 줄부터 N개의 줄에 차례로 주전자의 용량이 주어진다. N은 10000이하의 정수이고, K는 1,000,000이하의 정수이다. 막걸리의 용량은 231 -1 보다 작거나 같은 자연수 또는 0이다. 단, 항상 N ≤ K 이다. 즉, 주전자의 개수가 사람 수보다 많을 수는 없다.

출력

첫째 줄에 K명에게 나눠줄 수 있는 최대의 막걸리 용량 ml 를 출력한다.

---

풀이

아이디어

• 문제에 주어진 방식으로 입력을 받는다.
• 이분탐색을 1부터 가장 큰 값까지 돌아가면서 받는다, 아래 코드에서는 가장 큰 값은 lst[-1]이 보장되기 때문에 (정렬을 함.) end 값으로 lst[-1]을 주었음.
• 중앙값을 입력받은 lst에 나눠주어 k 라는 변수에 집어넣고 친구의 수와 비교하여 크거나 같으면 범위를 높이고, 낮으면 범위를 낮춘다.
• 단순 수학? 구현 문제이므로 아이디어가 어렵지 않다.

코드

import sys
input = sys.stdin.readline

def bi_sort(lst, start, end):
    while start <= end:
        mid = (start + end) // 2
        k= sum(map(lambda x: x//mid, lst))
        if k >= m:
            start = mid+1
        else:
            end = mid-1

    return end

n, m = map(int, input().split())
lst = [int(input()) for _ in range(n)]
lst.sort()

print(bi_sort(lst, 1, lst[-1]))

---

회고

조금 더 빨리 풀수 있을 거 같은데 아직 이분탐색의 범위 지정 방식이 조금 헷갈린다. 늘 그렇듯 몸빵을 하자. 많이 풀어보는게 답인것 같다. 실버문제로 이분탐색에 익숙해지고, 골드문제를 풀때까지 무한 반복해서 알고리즘에 익숙해지자.