Kulut

input/code.cpp: In function 'int main()':
input/code.cpp:50:22: warning: 'amt' may be used uninitialized in this function [-Wmaybe-uninitialized]
   50 |         p[neg.top()] += amt;
      |         ~~~~~~~~~~~~~^~~~~~

#include <bits/stdc++.h>
#define pb push_back
#define vi vector<int>
#define sz size
using namespace std;

int N, p[4], total;

int main() {
    ios_base::sync_with_stdio(false);

    cin >> N;

    for (int i = 0; i < N; i++) {
        char c;
        cin >> c;
        int s;
        cin >> s;
        p[c - 'A'] += s;
        total += s;
    }

    int target = total / 4;

    vector<pair<pair<char, char>, int>> ans;

    stack<int> pos, neg;

    for (int i = 0; i < 4; i++) {
        if (p[i] > target) pos.push(i);
        else if (p[i] < target) neg.push(i);
    }

    while (true) {
        pair<char,char> xd = make_pair('A' + neg.top(), 'A' + pos.top());
        int amt;

        int pdiff = abs(p[pos.top()] - target);
        int ndiff = abs(p[neg.top()] - target);

        if (pdiff == ndiff) {
            amt = pdiff;
        } else if (pdiff > ndiff) {
            amt = ndiff;
        } else if (pdiff < ndiff) {
            amt = pdiff;
        }

        p[pos.top()] -= amt;
        p[neg.top()] += amt;
        if (p[pos.top()] == target) pos.pop();
        if (p[neg.top()] == target) neg.pop();

        ans.pb(make_pair(xd, amt));

        if (pos.sz() == 0 && neg.sz() == 0) {
            break;
        }
    }

    cout << ans.sz() << endl;
    for (auto r : ans) {
        cout << r.first.first << " " << r.first.second << " " << r.second << endl;
    }

    return 0;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: RUNTIME ERROR