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