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

bool cmp(pair<int, int> x, pair<int, int> y)
{
    return x.first > y.first;
}

vector<int> solve(vector<int> A) 
{
    int n = A.size();
    vector<pair<int, int>> cA{};
    for(int i = 0; i < n; i++)
    {
        cA.emplace_back(A[i], i);
    }
    
    sort(cA.begin(), cA.end(), cmp);
    
    int sD = 0, sM = 0;
    vector<int> D{}, M{};
    for(int i = 0; i < n; i++)
    {
        if(sD > sM)
        {
            sM += cA[i].first;
            M.emplace_back(cA[i].second);
        }
        else
        {
            sD += cA[i].first;
            D.emplace_back(cA[i].second);
        }
    }

    // cout << abs(sD - sM) << "\n";

    for(int i = 0; i < D.size(); i++)
    {
        A[i] = D[i];
    }
    for(int i = 0; i < M.size(); i++)
    {
        A[i + D.size()] = M[i];
    }

    return A;
}
