백준 6236번 C++ 풀이

https://www.acmicpc.net/problem/6236

6236번: 용돈 관리

시간 제한	메모리 제한	solved.ac 티어
1초	128MB	실버 2

문제

현우는 용돈을 효율적으로 활용하기 위해 계획을 짜기로 하였다. 현우는 앞으로 N일 동안 자신이 사용할 금액을 계산하였고, 돈을 펑펑 쓰지 않기 위해 정확히 M번만 통장에서 돈을 빼서 쓰기로 하였다. 현우는 통장에서 K원을 인출하며, 통장에서 뺀 돈으로 하루를 보낼 수 있으면 그대로 사용하고, 모자라게 되면 남은 금액은 통장에 집어넣고 다시 K원을 인출한다. 다만 현우는 M이라는 숫자를 좋아하기 때문에, 정확히 M번을 맞추기 위해서 남은 금액이 그날 사용할 금액보다 많더라도 남은 금액은 통장에 집어넣고 다시 K원을 인출할 수 있다. 현우는 돈을 아끼기 위해 인출 금액 K를 최소화하기로 하였다. 현우가 필요한 최소 금액 K를 계산하는 프로그램을 작성하시오.

입력

1번째 줄에는 N과 M이 공백으로 주어진다. (1 ≤ N ≤ 100,000, 1 ≤ M ≤ N)

2번째 줄부터 총 N개의 줄에는 현우가 i번째 날에 이용할 금액이 주어진다. (1 ≤ 금액 ≤ 10000)

출력

첫 번째 줄에 현우가 통장에서 인출해야 할 최소 금액 K를 출력한다.

---

이 문제는 이진탐색으로 접근해야한다. 순차탐색으로 접근한다면, 탐색의 시작은 들어온 금액의 최댓값, 끝은 10억 (N = 10만, M = 1, 모든 금액이 1만일때)이고, 각 가격에서 10만개의 데이터를 계속 돌아야 하기 때문이다. 이는 당연히 시간 초과가 날 것이다.

 

또한 범위를 잘못생각했다가 알고리즘은 맞는데 이것때문에 자꾸 틀려서, 엄한 부분을 계속 수정했다.... 글을 쓰는 지금 생각이 드는 것은, 탐색의 끝값이 10억(N = 10만, M = 1, 모든 금액이 1만일때) 이어도 좋지만, 입력받은 모든 값의 합으로 해도 좋을 것 같다는 생각이 든다.

 

#include <bits/stdc++.h>
using namespace std;

int m, n;
int st = -1, ed = 1e9; // N = 10만, M = 1, 모든 금액이 1만일때 탐색의 끝값이 최대이다.
vector<int> v;

void input() {
    cin >> n >> m;
    v.resize(n);
    for(int& x : v) {
        cin >> x;
        st = max(x, st);
    }
}

bool check(int money) {
    int hand = money;
    int cnt = 1;

    for(int x : v) {
        if (money < x) return false;

        if (hand >= x) hand -= x;
        else {
            cnt++;
            hand = money - x;
        }
    }
    return cnt <= m;
}

void sol() { // 단순한 이진탐색
    int mid;
    int result;
    while (st <= ed) {
        mid = (st + ed) / 2;

        if (check(mid)) {
            ed = mid - 1;
            result = mid;
        }
        else {
            st = mid + 1;
        }
    }
    cout << result;
}

int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    input();
    sol();
}