백준 16112번 5차 전직 C++

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

16112번: 5차 전직

 

오름차순 정렬을 하면 경험치가 높은 퀘스트를 아케인스톤들에 더 나눠서 받을 수 있다.

 

#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;

vector<int64> v;

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

    int n, k;
    cin >> n >> k;
    v = vector<int64>(n);
    for (int i = 0; i < n; i++) {
        cin >> v[i];
    }
    sort(v.begin(), v.end());
    int cnt = 0;
    int64 sum = 0;
    for (int i = 0; i < n; i++) {
        sum += cnt * v[i];
        cnt = min(cnt + 1, k);
    }
    cout << sum;
}