취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 2512번 문제인 예산이다.
문제는 아래 링크를 확인하자.

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

요청하고 있는 예산이 가장 많은 예산요청들서부터 예산을 차근차근 깎아나가는 greedy한 전략으로 O(N) 시간 안에 문제를 해결할 수 있다.

이 문제를 greedy한 전략으로 문제를 해결할 경우, 초기 예산요청 그대로 예산을 할당할 수 있는 경우를 따로 처리해주는 것이 좋다.

아래는 제출한 소스코드이다.

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;

ll arr[10000];

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	ll mx = 0;
	ll total = 0;
	int N; cin >> N;
	for (int i = 0; i < N; i++) {
		ll x; cin >> x;
		total += x;
		arr[i] = x;
		mx = max(mx, x);
	}

	sort(arr, arr + N);
	ll M; cin >> M;
	
	if (total <= M) cout << mx;
	else {
		int i = N - 1;
		ll cnt = 0;
		for (; i > 0; i--) {
			cnt++;
			ll dist = arr[i] - arr[i - 1];
			total -= cnt * dist;
			if (total <= M) {
				cout << arr[i - 1] + (M - total) / cnt;
				break;
			}
		}
		if (i == 0) cout << M / N;
	}
}