백준 5545번 최고의 피자 C++

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

5545번: 최고의 피자

 

토핑의 열량들을 내림차순으로 정렬하고 큰거부터 하나씩 적용해보면서 1원당 열량이 줄어들기 전까지만 토핑을 추가한다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <list>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <limits.h>

using namespace std;
using int64 = long long; 

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

    int calory = 0;
    int price = 0;
    vector<int> calories;

    int n;
    cin >> n;
    int a, b;
    cin >> a >> b;
    price += a;
    int c;
    cin >> c;
    calory += c;
    for (int i = 0; i < n; i++) {
        int d;
        cin >> d;
        calories.push_back(d);
    }
    sort(calories.begin(), calories.end(), greater<>());
    for (int i = 0; i < n; i++) {
        int nextPrice = price + b;
        int nextCalory = calory + calories[i];
        if (calory / price > nextCalory / nextPrice) break;
        price = nextPrice;
        calory = nextCalory;
    }
    cout << calory / price;
}