Omenat

input/code.cpp: In function 'int main()':
input/code.cpp:25:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int64_t i=0;i<a.size();i++) {
                    ^

#include<iostream>
#include<vector>
#include<map>
#include<algorithm>
 
using namespace std;
 
typedef pair<int64_t,int64_t> P;
int main(void) {
  int64_t n;
  cin >> n;
 
  vector<P> a(n);
  for(int64_t i=0;i<n;i++) {
    cin>>a[i].first;
    a[i].first=-a[i].first;
    a[i].second=i;
  }
 
  sort(a.begin(), a.end());
 
  map<int64_t,int> who;
 
  int64_t A=0,B=0;
  for(int64_t i=0;i<a.size();i++) {
    if(A <= B) {
      A+=-a[i].first;
      who[a[i].second]=1;
    } else {
      B+=-a[i].first;
      who[a[i].second]=2;
    }
  }
 
  for(int64_t i=0;i<n;i++) {
    cout << who[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