Aitaus

input/code.cpp: In function 'int main()':
input/code.cpp:21:35: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i = 0; i < v.size(); i++){
                                   ^

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

int main()
{
    int n;
    cin >> n;

    vector<int> v(n);
    int l = 0;
    for(int i = 0; i < n; i++){
        cin >> v[i];
        l += v[i];
    }


    int cost = 0;
    while(v.size() > 1){
        int avg = l / v.size() * 2;
        sort(v.begin(), v.end());
        for(int i = 0; i < v.size(); i++){
            int num = v[i];
            int missing = avg - num;
            if(missing <= 0){
                continue;
            }
            auto it = lower_bound(v.begin() + i + 1, v.end(), missing);
            if(it == v.end())
                it--;
            if(it != v.begin() && it != v.begin() + i + 1){
                int diff1 = abs(missing - *it);
                int diff2 = abs(missing - *(it - 1));
                if(diff2 < diff1)
                    it--;
            }
            if(it == v.begin() + i)
                continue;
            int num2 = *it;

            v.erase(it);
            v[i] += num2;
            cost += v[i];
        }
    }

    cout << cost << "\n";
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 1

Verdict: ACCEPTED

Test 3

Group: 1

Verdict: WRONG ANSWER

Test 4

Group: 1

Verdict: WRONG ANSWER

Test 5

Group: 1

Verdict: WRONG ANSWER

Test 6

Group: 2

Verdict: ACCEPTED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: WRONG ANSWER

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 11

Group: 3

Verdict: ACCEPTED

Test 12

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 3

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 3

Verdict: TIME LIMIT EXCEEDED