CSES - NOI 2019 Open - Results
Submission details
Task:Thieves and Prisons
Sender:jakob_nogler
Submission time:2019-03-11 19:43:22 +0200
Language:C++
Status:READY
Result:13
Feedback
groupverdictscore
#10
#2ACCEPTED13
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.02 s2, 4, 5details
#2ACCEPTED0.01 s2, 4, 5details
#3ACCEPTED0.02 s2, 4, 5details
#4ACCEPTED0.04 s2, 4, 5details
#5ACCEPTED0.02 s2, 4, 5details
#6ACCEPTED0.02 s4, 5details
#7ACCEPTED0.02 s4, 5details
#8ACCEPTED0.01 s4, 5details
#9ACCEPTED0.03 s1, 3, 4, 5details
#10ACCEPTED0.01 s1, 3, 4, 5details
#11ACCEPTED0.03 s1, 3, 4, 5details
#12ACCEPTED0.03 s1, 3, 4, 5details
#13--1, 3, 4, 5details
#14--1, 3, 4, 5details
#15ACCEPTED0.01 s1, 3, 4, 5details
#16ACCEPTED0.02 s1, 3, 4, 5details
#17ACCEPTED0.02 s1, 2, 3, 4, 5details
#18ACCEPTED0.01 s1, 3, 4, 5details
#19ACCEPTED0.10 s2, 5details
#20ACCEPTED0.09 s2, 5details
#21ACCEPTED0.10 s2, 5details
#22ACCEPTED0.10 s5details
#230.10 s5details
#240.03 s3, 4, 5details
#250.03 s3, 4, 5details
#260.02 s3, 4, 5details
#270.03 s3, 4, 5details
#28ACCEPTED0.02 s4, 5details
#29ACCEPTED0.02 s4, 5details
#30ACCEPTED0.04 s4, 5details
#31ACCEPTED0.03 s4, 5details
#32ACCEPTED0.02 s2, 4, 5details
#33ACCEPTED0.02 s2, 4, 5details
#34ACCEPTED0.02 s2, 4, 5details
#35ACCEPTED0.02 s2, 4, 5details
#360.12 s3, 5details
#370.12 s3, 5details
#380.12 s3, 5details
#390.12 s3, 5details
#40ACCEPTED0.11 s5details
#41ACCEPTED0.11 s5details
#42ACCEPTED0.11 s5details
#43ACCEPTED0.12 s5details
#44ACCEPTED0.10 s2, 5details
#45ACCEPTED0.11 s2, 5details
#46ACCEPTED0.12 s2, 5details
#47ACCEPTED0.12 s2, 5details

Code

#include <bits/stdc++.h>
#define pb push_back
#define fi first
#define se second

using namespace std;

typedef pair<int, int> pii;
typedef vector<int> vi;
typedef vector<pii> vpii;

const int N = 100005;

vpii event[N]; //second true if he has been caught
bool sol = true;
int ans[N];
vi open;

struct seg{
    int a, b, id;
    bool operator < (const seg &o) const{
        return b > o.b; 
    }
};

bool cmp(const seg &lhs, const seg &rhs){
    return lhs.a < rhs.a;
}

void display(const seg &v){
    cout << v.a << " " << v.b << " " << v.id << endl;
}

vector<seg> vec;
priority_queue<seg> pq;

int main(){
    int n, k, m, x; char t;
    cin >> n >> k >> m;
    for(int i = 0; i < m; i++){
        cin >> t >> x;
        if(t == 'C') event[x].pb({i, true}); 
        else{
            event[x].pb({i, false});
            open.pb(i);
            ans[i] = open.size(); //TO CHANGE
        }
    }

    for(int i = 1; i <= n; i++){
        for(int j = 0; j < int(event[i].size()) - 1; j++)
            if(event[i][j].se)
                vec.pb({event[i][j].fi + 1, event[i][j + 1].fi - 1, i});
        if(!event[i].empty() && event[i].back().se) vec.pb({event[i].back().fi + 1, m + 1, i});
    }

    sort(vec.begin(), vec.end(), cmp);


    //for(auto u : vec) display(u);

    int idx = 0;
    for(int i = 0; i < int(open.size()) && sol; i++){
        //cout << i << ", " << open[i] << ": " << endl;
        while(idx < int(vec.size()) && vec[idx].a <= open[i]) pq.push(vec[idx++]);
        int cnt = 0;

        //cout << "top: ";
        //display(pq.top());
        //cout << "size " << pq.size() << endl;

        if(pq.empty() || pq.top().b < open[i]){sol = false; break;}
        while((i == int(open.size()) - 1 && !pq.empty()) || pq.top().b < open[i + 1]){
            cnt++;
            ans[pq.top().a - 1] = i + 1;
            pq.pop();
        }
        if(!sol || (cnt == 0 && pq.empty())){sol = false; break;}

        //cout << ": " << cnt << endl;
        //cout << "size " << pq.size() << endl;

        if(cnt == 0){
            ans[pq.top().a - 1] = i + 1;
            pq.pop();
        }

    }
    
    while(!pq.empty()){
        if(pq.top().b != m + 1) sol = false;
        ans[pq.top().a - 1] = 1;
        pq.pop();
    }
    for(int i = idx; i < int(vec.size()); i++){
        if(vec[i].b != m + 1) sol = false;
        ans[vec[i].a - 1] = 1;
    }

    if(!sol){
        cout << "IMPOSSIBLE\n";
        return 0;
    }

    for(int i = 0; i < m; i++)
        cout << ans[i] << " ";
    cout << endl;
}

Test details

Test 1

Group: 2, 4, 5

Verdict: ACCEPTED

input
1 1 1
C 1

correct output

user output

Test 2

Group: 2, 4, 5

Verdict: ACCEPTED

input
1 1 1
O 1

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 3

Group: 2, 4, 5

Verdict: ACCEPTED

input
1 1 2
C 1
C 1

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 4

Group: 2, 4, 5

Verdict: ACCEPTED

input
1 1 2
C 1
O 1

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 5

Group: 2, 4, 5

Verdict: ACCEPTED

input
1 1 2
O 1
C 1

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 6

Group: 4, 5

Verdict: ACCEPTED

input
2 1 2
C 1
C 2

correct output
1 1 

user output
1 1 

Test 7

Group: 4, 5

Verdict: ACCEPTED

input
2 1 2
C 1
O 1

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 8

Group: 4, 5

Verdict: ACCEPTED

input
2 1 2
C 1
O 2

correct output
1 1 

user output
1 1 

Test 9

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 5
C 1
C 2
O 3
C 1
...

correct output
1 1 1 1 1 

user output
1 1 1 1 1 

Test 10

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 5
C 1
C 2
O 3
O 3
...

correct output
2 1 2 1 1 

user output
1 2 1 2 1 

Test 11

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 5
C 1
C 2
O 3
O 1
...

correct output
2 1 2 1 1 

user output
1 2 1 2 1 

Test 12

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 5
C 1
C 2
O 1
O 3
...

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 13

Group: 1, 3, 4, 5

Verdict:

input
3 2 4
C 1
O 2
C 1
O 3

correct output
1 1 1 1 

user output
(empty)

Test 14

Group: 1, 3, 4, 5

Verdict:

input
3 2 4
C 1
O 2
C 2
O 1

correct output
1 1 1 1 

user output
(empty)

Test 15

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 3
C 1
C 2
C 3

correct output
1 1 1 

user output
1 1 1 

Test 16

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
3 2 3
O 1
C 2
C 3

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 17

Group: 1, 2, 3, 4, 5

Verdict: ACCEPTED

input
2 2 7
C 1
O 2
O 2
O 2
...

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

Test 18

Group: 1, 3, 4, 5

Verdict: ACCEPTED

input
4 2 5
C 2
O 3
C 1
O 4
...

correct output
1 1 1 1 1 

user output
1 1 2 2 1 

Test 19

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 1
C 2
C 3
C 4
...

correct output
50000 49999 49998 49997 49996 ...

user output
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

Test 20

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 1
C 2
C 3
C 4
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 21

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 1
C 2
C 3
C 4
...

correct output
20000 20000 20000 20000 20000 ...

user output
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

Test 22

Group: 5

Verdict: ACCEPTED

input
100000 100 100000
C 1
C 2
C 3
C 4
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

Test 23

Group: 5

Verdict:

input
100000 99 100000
C 1
C 2
C 3
C 4
...

correct output
IMPOSSIBLE

user output
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

Test 24

Group: 3, 4, 5

Verdict:

input
500 2 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 25

Group: 3, 4, 5

Verdict:

input
500 2 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 9 8 7 ...

Test 26

Group: 3, 4, 5

Verdict:

input
500 2 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 2 1 2 ...

user output
1 1 2 2 5 4 3 3 8 4 160 5 7 16...

Test 27

Group: 3, 4, 5

Verdict:

input
500 2 500
C 384
O 62
C 387
C 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 83 71 72 96 104 3 2 45 6 1...

Test 28

Group: 4, 5

Verdict: ACCEPTED

input
500 250 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 29

Group: 4, 5

Verdict: ACCEPTED

input
500 250 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 2 1 3 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 9 8 7 ...

Test 30

Group: 4, 5

Verdict: ACCEPTED

input
500 250 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 3 2 3 3 2 2 2 5 4 2 ...

user output
1 1 2 2 7 3 4 3 10 4 5 5 6 8 9...

Test 31

Group: 4, 5

Verdict: ACCEPTED

input
500 250 500
C 384
O 62
C 387
C 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 8 23 3 2 109 7 17 4 109 10...

Test 32

Group: 2, 4, 5

Verdict: ACCEPTED

input
500 500 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 33

Group: 2, 4, 5

Verdict: ACCEPTED

input
500 500 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 2 1 3 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 8 9 7 ...

Test 34

Group: 2, 4, 5

Verdict: ACCEPTED

input
500 500 500
C 384
O 62
C 387
O 473
...

correct output
1 1 1 1 2 1 3 3 3 2 2 2 2 4 5 ...

user output
1 1 2 2 4 8 3 3 9 4 5 5 10 7 6...

Test 35

Group: 2, 4, 5

Verdict: ACCEPTED

input
500 500 500
C 384
O 62
C 387
C 473
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 109 109 109 2 4 41 109 109...

Test 36

Group: 3, 5

Verdict:

input
100000 2 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 37

Group: 3, 5

Verdict:

input
100000 2 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 2 2 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 8 7 41...

Test 38

Group: 3, 5

Verdict:

input
100000 2 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 ...

user output
1 1 2 2 3 32504 5 3 4 4 32504 ...

Test 39

Group: 3, 5

Verdict:

input
100000 2 100000
C 89384
O 54062
C 85387
C 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 6399 8750 18880 16431 1888...

Test 40

Group: 5

Verdict: ACCEPTED

input
100000 50000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 41

Group: 5

Verdict: ACCEPTED

input
100000 50000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 3 2 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 8 9 ...

Test 42

Group: 5

Verdict: ACCEPTED

input
100000 50000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 3 2 3 3 3 3 3 3 4 5 ...

user output
1 1 2 2 9 3 8 3 4 4 5 5 10 7 6...

Test 43

Group: 5

Verdict: ACCEPTED

input
100000 50000 100000
C 89384
O 54062
C 85387
C 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 8 20160 20160 1887 1297 ...

Test 44

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 7 8 ...

Test 45

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 3 2 1 ...

user output
1 1 2 2 3 3 4 4 5 5 6 6 7 8 9 ...

Test 46

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 89384
O 54062
C 85387
O 53318
...

correct output
1 1 1 1 2 1 3 3 3 3 3 3 4 5 3 ...

user output
1 1 2 2 8 9 3 3 4 4 5 5 7 6 10...

Test 47

Group: 2, 5

Verdict: ACCEPTED

input
100000 100000 100000
C 89384
O 54062
C 85387
C 53318
...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
1 1 2339 3482 20160 3 8402 5 2...