[백준/C++] 13702번. 이상한 술집

[백준/C++] 13702번. 이상한 술집

1. 문제

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

2. 입력

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

3. 출력

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

4. 풀이

  • 이분 탐색 문제
  • 이분 탐색으로 적절한 값을 찾는다.
  • start를 0으로 end를 2147483647(2^31 - 1)로 설정했다.
  • mid의 의미는 나눠주는 막걸리의 용량이다.
  • 각 막걸리에 담긴 용량을 mid로 나눈 몫을 더한 값 result를 구한다.
    mid만큼 사람들에게 나누어주면 result명에게 나누어 줄 수 있다는 의미
  • result가 k보다 작으면 충분히 나누어줄 수 없기 때문에 나누어주는 막걸리의 양을 줄여야 한다. (→ end = mid - 1)
  • result가 k보다 크면 k명에게 나누어 주었기 때문에 answer에 해당 값을 저장하고 막걸리 양을 늘려서 더 나눠줄 수 있는지 확인한다. (→ start = mid + 1)

5. 처음 코드와 달라진 점

  • 2^31-1이라서 int 범위로 해결할 수 있을 줄 알았는데 안 된다.
  • long long으로 자료형을 변경해주고 통과했다.
    이럴 때는 파이썬으로 할까 싶다

6. 코드

#include <iostream>
#include <vector>
#define INF 2147483647

using namespace std;

int n, k;
int arr[10000];

int main() {
  cin.tie(NULL);
  ios_base::sync_with_stdio(false);

  cin >> n >> k;
  for (int i = 0; i < n; i++) {
    cin >> arr[i];
  }

  int start = 0;
  long long end = INF;
  long long answer;

  while (start <= end) {
    long long mid = (start + end) / 2;
    long long result = 0;

    for (int i = 0; i < n; i++) {
      result += arr[i] / mid;
    }

    if (result < k) { // 나눌 수 없음
      end = mid - 1;
    }
    else {
      answer = mid;
      start = mid + 1;
    }
  }

  cout<<answer;
}

좋은 웹페이지 즐겨찾기