| Task: | Omenat |
| Sender: | Henrik Lievonen |
| Submission time: | 2015-12-20 17:01:06 +0200 |
| Language: | C++ |
| Status: | READY |
| Result: | 31 |
| group | verdict | score |
|---|---|---|
| #1 | ACCEPTED | 30.6 |
| test | verdict | time | score | |
|---|---|---|---|---|
| #1 | ACCEPTED | 0.78 s | 2.4 | details |
| #2 | ACCEPTED | 0.73 s | 4.1 | details |
| #3 | ACCEPTED | 0.83 s | 3 | details |
| #4 | ACCEPTED | 0.85 s | 3.8 | details |
| #5 | ACCEPTED | 0.75 s | 2.6 | details |
| #6 | ACCEPTED | 0.78 s | 2.9 | details |
| #7 | ACCEPTED | 0.74 s | 3.2 | details |
| #8 | ACCEPTED | 0.78 s | 3 | details |
| #9 | ACCEPTED | 0.87 s | 3.4 | details |
| #10 | ACCEPTED | 0.81 s | 2.2 | details |
Code
// G
#include <iostream>
#include <random>
#include <vector>
#include <algorithm>
using namespace std;
typedef unsigned long long int ull;
mt19937 mt(0);
uniform_int_distribution<> flip(1, 2);
vector<pair<ull, int>> apples;
void print(const vector<char> &r) {
for (char c : r)
cout << c << '\n';
}
int main() {
int n;
cin >> n;
for (int i = 0; i < n; i++) {
ull a;
cin >> a;
apples.push_back(make_pair(a, i));
}
vector<char> naive(n);
ull naived;
{
ull u = 0;
ull m = 0;
for (int i = 0; i < n; i++) {
auto a = apples[i];
if (u > m) {
naive[i] = '2';
m += a.first;
} else {
naive[i] = '1';
u += a.first;
}
}
naived = min(u-m, m-u);
}
vector<char> sorted(n);
ull sortedd = 1ull<<60;
{
sort(apples.begin(), apples.end());
ull u = 0;
ull m = 0;
for (int i = n-1; i >= 0; i--) {
auto a = apples[i];
if (u > m) {
sorted[a.second] = '2';
m += a.first;
} else {
sorted[a.second] = '1';
u += a.first;
}
}
sortedd = min(u-m, m-u);
}
vector<char> bitthing(n);
ull bitthingd = 1ull<<60;
{
const int bits = min(n, 21);
for (int k = 0; k < (1<<bits); k++) {
ull u = 0;
ull m = 0;
int i = n-1;
for (int j = 0; j < bits; j++, i--) {
auto a = apples[i];
if (k & (1<<j)) {
sorted[a.second] = '2';
m += a.first;
} else {
sorted[a.second] = '1';
u += a.first;
}
}
for (; i >= 0; i--) {
auto a = apples[i];
if (u > m) {
sorted[a.second] = '2';
m += a.first;
} else {
sorted[a.second] = '1';
u += a.first;
}
}
sortedd = min(u-m, m-u);
}
}
if (naived < sortedd && naived < bitthingd) {
print(naive);
} else if (sortedd < naived && sortedd < bitthingd) {
print(sorted);
} else if (bitthingd < naived && bitthingd < sortedd) {
print(bitthing);
} else {
print(sorted);
}
}Test details
Test 1
Verdict: ACCEPTED
| input |
|---|
| 95 779724552 231968220 985023789 ... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 1 1 2 1 2 ... |
Test 2
Verdict: ACCEPTED
| input |
|---|
| 85 229722261 51722691 862338862 8... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 1 2 2 1 2 ... |
Test 3
Verdict: ACCEPTED
| input |
|---|
| 97 398995377 989444445 634573915 ... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 1 2 1 1 1 ... |
Test 4
Verdict: ACCEPTED
| input |
|---|
| 99 843687873 164010938 51269970 4... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 2 1 2 1 2 ... |
Test 5
Verdict: ACCEPTED
| input |
|---|
| 90 864611617 418460939 773297829 ... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 2 2 1 1 2 ... |
Test 6
Verdict: ACCEPTED
| input |
|---|
| 92 289890246 25801423 763027596 7... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 2 1 1 2 2 ... |
Test 7
Verdict: ACCEPTED
| input |
|---|
| 89 879039800 50522278 850785072 4... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 2 2 2 1 1 ... |
Test 8
Verdict: ACCEPTED
| input |
|---|
| 96 27192469 222283781 681532515 1... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 1 2 1 1 1 ... |
Test 9
Verdict: ACCEPTED
| input |
|---|
| 100 186459081 254674429 394007236 ... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 2 2 1 1 1 ... |
Test 10
Verdict: ACCEPTED
| input |
|---|
| 98 612168861 979831717 671087051 ... |
| correct output |
|---|
| (empty) |
| user output |
|---|
| 1 2 1 1 2 ... |
