#include "partition.h"
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;

struct st_aux{
    int a,ind;
};

vector<st_aux> aux;

bool cmp(st_aux a,st_aux b){
    return a.a<b.a;
}

vector<int> solve(vector<int> A) {
    int min1=1e9,mask=0,bit,i,auxmin1,cnt,sum=0,sumcur;
    aux.clear();
    for(auto it:A) sum+=it;
    ///cout<<sum<<" ||\n";
    for(bit=0;bit<(1<<(A.size()));bit++){
        auxmin1=cnt=0;sumcur=0;
        for(i=0;i<A.size();i++){
            if(bit&(1<<i)){
                sumcur+=A[i];
                auxmin1=max(auxmin1,A[i]);
                cnt++;
            }
        }
        ////cout<<bit<<": "<<sumcur<<" "<<sumcur-auxmin1<<" "<<max((((int)A.size()-cnt)-cnt),cnt-(((int)A.size()-cnt)))<<"|\n";
        if(2*(sumcur-auxmin1)<sum && 2*sumcur>=sum && max((((int)A.size()-cnt)-cnt),cnt-(((int)A.size()-cnt)))<min1){
            mask=bit;
            min1=max((((int)A.size()-cnt)-cnt),cnt-(((int)A.size()-cnt)));
        }
    }
    ///cout<<"| "<<mask<<"| ";
    for(i=0;i<A.size();i++){
        if((1<<i)&mask){
            aux.push_back({A[i],i});
        }
    }
    vector<int> rasp;rasp.clear();
    sort(aux.begin(),aux.end(),cmp);
    for(auto it:aux) rasp.push_back(it.ind);
    for(i=0;i<A.size();i++){
        if((1<<i)&mask) continue;
        rasp.push_back(i);
    }
    return rasp;
}
