Tikut

input/code.cpp: In function 'int sumArr(std::vector<int>)':
input/code.cpp:11:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   11 |     for (int i = 0; i < arr.size(); i++) {
      |                     ~~^~~~~~~~~~~~

#include <algorithm>
#include <cmath>
#include <iostream>
#include <vector>
#include <climits>

using namespace std;

int sumArr(vector<int> arr) {
    int r = 0;
    for (int i = 0; i < arr.size(); i++) {
        r += arr[i];
    }

    return r;
}

int solve(vector<vector<int>> v, int n, int k, int mn) {
    for (int i = 0; i < k; i++) {
        vector<int> v1 = v[v.size()-1];
        int parent = sumArr(v1);
        int divider = v1.size();

        v.pop_back();

        divider++;

        vector<int> temp(parent % divider, (parent / divider) + 1);

        vector <int> temp2(divider - (parent % divider), (parent / divider));

        if ((parent / divider) < mn) {
            mn = parent / divider;
        }

        for (int u : temp2) {
            temp.push_back(u);
        }

        v.push_back(temp);

        sort(v.begin(), v.end());

        int mx = v[v.size()-1][0];
        cout << mx - mn << " ";
    }
    return 0;
}

int main() {
    int n,k ;
    int mn = INT_MAX;
    cin >> n >> k;

    vector<vector<int>> v;
    for (int i = 0; i < n; i++) {
        vector<int> temp;
        int a;
        cin >> a;
        temp.push_back(a);
        if (temp[0] < mn) {
            mn = temp[0];
        }
        v.push_back(temp);
    }
    sort(v.begin(), v.end());
    solve(v, n, k, mn);

    return 0;
}

Test 1

Group: 1, 3, 4, 5, 6

Verdict: ACCEPTED

Test 2

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 3

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 4

Group: 1, 4, 5, 6

Verdict: ACCEPTED

Test 5

Group: 2, 5, 6

Verdict: ACCEPTED

Test 6

Group: 2, 5, 6

Verdict: ACCEPTED

Test 7

Group: 3, 5, 6

Verdict: ACCEPTED

Test 8

Group: 3, 5, 6

Verdict: ACCEPTED

Test 9

Group: 3, 5, 6

Verdict: ACCEPTED

Test 10

Group: 3, 5, 6

Verdict: ACCEPTED

Test 11

Group: 3, 5, 6

Verdict: WRONG ANSWER

Test 12

Group: 4, 5, 6

Verdict: ACCEPTED

Test 13

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 14

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 15

Group: 4, 5, 6

Verdict: WRONG ANSWER

Test 16

Group: 5, 6

Verdict: ACCEPTED

Test 17

Group: 5, 6

Verdict: WRONG ANSWER

Test 18

Group: 5, 6

Verdict: WRONG ANSWER

Test 19

Group: 5, 6

Verdict: WRONG ANSWER

Test 20

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 6

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 6

Verdict: TIME LIMIT EXCEEDED