CSES - HIIT Open 2018 - Results
Submission details
Task:Coins
Sender:HIIT AND RUN
Submission time:2018-05-26 15:37:38 +0300
Language:C++
Status:READY
Result:
Test results
testverdicttime
#10.67 sdetails
#2ACCEPTED0.45 sdetails
#3ACCEPTED0.43 sdetails
#4ACCEPTED0.44 sdetails
#50.61 sdetails

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:62:29: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         if (isojaMatchaavia >= from->size()) ans.push_back(to == &r ? "<" : ">");
             ~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int main() {
int n;
cin >> n;
set<ll> l;
set<ll> r;
vector<ll> backtrackActions;
for (int i=0; i< n; i++) {
ll coin;
cin >> coin;
ll leftOrRight;
cin >> leftOrRight;
backtrackActions.push_back(coin);
backtrackActions.push_back(leftOrRight);
if (leftOrRight == 1) l.insert(coin);
else r.insert(coin);
}
auto to = &r;
auto from = &l;
if (!l.empty()) {
if (r.empty() || *l.rbegin() > *r.rbegin()) {
to = &l;
from = &r;
}
}
ll largest = *to->rbegin();
vector<string> ans;
set<ll> pq;
ll countRemovedSmalls = 0;
ll i = backtrackActions.size() -1;
while (true) {
ll smallestInFrom = (from->empty() ? 0 : *from->begin());
//cout << "largest" << largest << "smallest" << smallestInFrom << endl;
//cout << "from" << from->size() << endl;
ll countSmallerThanFromSmallest = max((ll) 0, smallestInFrom - 1 - countRemovedSmalls);
ll isojaMatchaavia = to->size() - countSmallerThanFromSmallest;
//cout << "isoja" << isojaMatchaavia << endl;
if (isojaMatchaavia >= from->size()) ans.push_back(to == &r ? "<" : ">");
else ans.push_back("?");
if (i <= 2) break;
ll leftOrRight = backtrackActions[i--];
ll coin = backtrackActions[i--];
if (coin < smallestInFrom) {
to->erase(coin);
countRemovedSmalls++;
} else if (coin == smallestInFrom) {
from->erase(coin);
countRemovedSmalls++;
ll newSmallest = (from->empty() ? 0 : *from->begin());
while (!pq.empty() && *pq.begin() < newSmallest) {
pq.erase(pq.begin());
countRemovedSmalls++;
}
} else if (coin == largest) {
to->erase(coin);
largest = (to->empty() ? -1 : *to->rbegin());
//System.out.println("largest " + largest);
if (largest == -1 || (!from->empty() && *from->rbegin() > largest)) {
// swap to and from
auto helper = to;
to = from;
from = helper;
largest = *to->rbegin();
}
pq.insert(coin);
} else {
if (leftOrRight == 2) r.erase(coin);
else l.erase(coin);
pq.insert(coin);
}
}
for (int i=ans.size()-1; i >= 0; i--) {
cout << ans[i] << endl;
}
}

Test details

Test 1

Verdict:

input
200000
175878 1
146174 1
4939 2
181388 1
...

correct output
>
>
>
>
>
...

user output
>
>
>
>
>
...

Test 2

Verdict: ACCEPTED

input
200000
1 2
2 1
3 2
4 1
...

correct output
<
>
<
>
<
...

user output
<
>
<
>
<
...

Test 3

Verdict: ACCEPTED

input
200000
1 1
2 1
3 1
4 1
...

correct output
>
>
>
>
>
...

user output
>
>
>
>
>
...

Test 4

Verdict: ACCEPTED

input
200000
1 1
2 1
3 1
4 1
...

correct output
>
>
>
>
>
...

user output
>
>
>
>
>
...

Test 5

Verdict:

input
200000
188909 2
58944 1
26824 1
143263 2
...

correct output
<
<
?
<
<
...

user output
<
<
?
<
<
...