Algorithm
After the educational reform Polycarp studies only two subjects at school, Safety Studies and PE (Physical Education). During the long months of the fourth term, he received n marks in them. When teachers wrote a mark in the journal, they didn't write in what subject the mark was for, they just wrote the mark.
Now it's time to show the journal to his strict parents. Polycarp knows that recently at the Parent Meeting the parents were told that he received a Safety Studies marks and b PE marks (a + b = n). Now Polycarp wants to write a subject's name in front of each mark so that:
- there are exactly a Safety Studies marks,
- there are exactly b PE marks,
- the total average score in both subjects is maximum.
An average subject grade is the sum of all marks in it, divided by the number of them. Of course, the division is performed in real numbers without rounding up or down. Polycarp aims to maximize the x1 + x2, where x1 is the average score in the first subject (Safety Studies), and x2 is the average score in the second one (Physical Education).
The first line contains an integer n (2 ≤ n ≤ 105), n is the number of marks in Polycarp's Journal. The second line contains two positive integers a, b (1 ≤ a, b ≤ n - 1, a + b = n). The third line contains a sequence of integers t1, t2, ..., tn (1 ≤ ti ≤ 5), they are Polycarp's marks.
Print the sequence of integers f1, f2, ..., fn, where fi (1 ≤ fi ≤ 2) is the number of a subject to which the i-th mark should be attributed. If there are several possible solutions, then print such that the sequence f1, f2, ..., fn is the smallest lexicographically.
The sequence p1, p2, ..., pn is lexicographically less than q1, q2, ..., qn if there exists such j (1 ≤ j ≤ n) that pi = qi for all 1 ≤ i < j, аnd pj < qj.
5
3 2
4 4 5 4 4
1 1 2 1 2
4
2 2
3 5 4 5
1 1 2 2
6
1 5
4 4 4 5 4 4
2 2 2 1 2 2
In the first sample the average score in the first subject is equal to 4, and in the second one — to 4.5. The total average score is 8.5.
Code Examples
#1 Code Example with C++ Programming
Code -
C++ Programming
#include <cstdio>
#include <iostream>
#include <vector>
#include <algorithm>
int main(){
long n; scanf("%ld", &n);
long a, b; scanf("%ld %ld", &a, &b);
std::vector<std::pair<long, long> > f(n); for(long p = 0; p < n; p++){scanf("%ld", &f[p].first); f[p].second = p;}
if(a > b){sort(f.begin(), f.end());}
else if(a < b){sort(f.rbegin(), f.rend());}
std::vector < long> s(n);
for(long p = 0; p < n; p++){s[f[p].second] = 1 + (p >= a);}
for(long p = 0; p < n; p++){printf("%ld ", s[p]);}
puts("");
return 0;
}Input
3 2
4 4 5 4 4
Output
Demonstration
Codeforces Solution-C. Average Score-Solution in C, C++, Java, Python