Omenat

#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <unordered_set>
#include <stdio.h>
#include <string.h>
#include <unordered_map>
#include <fstream>
#include <set>
#include <map>
 
#define MOD 1000000007
#define ll long long
//#define N (1<<50)
#define float double
#define str string
 
using namespace std;
 
struct lol{
    vector<int> o1,o2;
    ll diff;
};
 
bool operator<(lol a, lol b){
    return a.diff<b.diff;
}
 
lol op(lol a, lol b){
    if(a<b)
        return op(b,a);
    lol c;
    c.diff=a.diff-b.diff;
 
    for(auto asd:a.o1)
        c.o1.push_back(asd);
    for(auto asd:a.o2)
        c.o2.push_back(asd);
    for(auto asd:b.o1)
        c.o2.push_back(asd);
    for(auto asd:b.o2)
        c.o1.push_back(asd);
 
    return c;
}
 
int vast[100];
 
int main(){
    int n;
    cin>>n;
 
    priority_queue<lol> pq;
 
    for(int i=0; i<n; i++){
        lol uus;
        ll w;
        cin>>w;
        uus.diff=w;
        uus.o1.push_back(i);
        pq.push(uus);
    }
 
    while(pq.size()>=2){
        lol a,b;
        a=pq.top();
        pq.pop();
        b=pq.top();
        pq.pop();
 
        pq.push(op(a,b));
    }
 
    for(int i:pq.top().o1)
        vast[i]=1;
    for(int i:pq.top().o2)
        vast[i]=2;
 
    for(int i=0; i<n; i++)
        cout<<vast[i]<<" ";
    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