Tikut

input/code.cpp: In function 'int calculateMinDifferenceLow(const std::vector<int>&, int)':
input/code.cpp:9:10: warning: variable 'ones' set but not used [-Wunused-but-set-variable]
    9 |     bool ones = false;
      |          ^~~~

#include <bits/stdc++.h>
#define ln "\n"
using namespace std;
using ll = long long;
int n, m;

int calculateMinDifferenceLow(const vector<int>& vec, int k) {
    int thers = 0, twos = 0;
    bool ones = false;
    for(int i = 0; i < n; i++) {
        if(vec[i] == 2) {
            twos++;
        } else if (vec[i] == 3) {
            thers++;
        } else {
            ones = true;
        }
    }

    if(k >= twos+thers*2) {
        return 0;
    } else if(thers <= k) {
        return 1;
    } else {
        return 2;
    }
}

void otherCalculaiton (vector<int>& vec) {
        multiset<int> s;
        for(int i = 0; i < n; i++) {
            s.insert(vec[i]);
        } 
        for(int i = 1; i <= m; i++) {
        multiset<int> temp = s;
        int best = INT_MAX;
        int target = *temp.rbegin();
 
        while(target > 0) {
            int cuts = i;
            multiset<int> final;
 
            for(auto it = temp.rbegin(); it != temp.rend(); it++) {
 
                if(*it > target && cuts > 0) {
                    int left = 0;
 
                    for(int j = 2; j <= cuts+1; j++) {
                        int val = *it/j;
                        bool done = false;
                        left = *it % j;
                        int add = 0;
                        if(left > 0) add++;
                        if(val + add <= target || j == cuts+1) {
                            done = true;
                            cuts -= (j-1);
                            vector<int> base;
                            for(int v = 0; v < j; v++) base.push_back(val);
 
                            while(left > 0) {
                                for(int v = 0; v < (j+1);v++) {
                                    base[v]++;
                                    left--; 
                                    if(left == 0) break;
                                }
                            }
 
                            for(int g:base) {
                                final.insert(g);
                            }
                        }
                        if(done) break;
                    }
                } else {
 
                    final.insert(*it);
                }
            }
 
            if(cuts == 0) {
                best = min(best,*final.rbegin()-*final.begin());
            } else {
                for(int j =1 ; j <= cuts; j++) {
                    auto biggest = final.end();
                    biggest--;
                    int val = *biggest;
                    final.erase(biggest);
                    final.insert(val/2);
                    if(val % 2 == 1) final.insert((val/2)+1);
                    else final.insert(val/2);
                }
                best = min(best,*final.rbegin()-*final.begin());
 
            }
            target--;
        }
        cout << best << ln;
    }

}

int main() {
    cin >> n >> m;
    vector<int> vec(n);
    for (int i = 0; i < n; ++i) {
        cin >> vec[i];
    }
    if(m <= 2) {
        otherCalculaiton(vec);
    } else {
        for (int k = 1; k <= m; ++k) {
            cout << calculateMinDifferenceLow(vec, k) << " ";
        }
    }

}

Test 1

Group: 1, 3, 4, 5, 6

Verdict: ACCEPTED

Test 2

Group: 1, 4, 5, 6

Verdict: WRONG ANSWER

Test 3

Group: 1, 4, 5, 6

Verdict: WRONG ANSWER

Test 4

Group: 1, 4, 5, 6

Verdict: WRONG ANSWER

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: ACCEPTED

Test 12

Group: 4, 5, 6

Verdict: WRONG ANSWER

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: WRONG ANSWER

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