CSES - NOI 2019 - Results
Submission details
Task:Thieves and Prisons
Sender:Santeri Toivonen
Submission time:2019-03-06 13:33:06 +0200
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.03 s2, 4, 5details
#2ACCEPTED0.03 s2, 4, 5details
#3ACCEPTED0.03 s2, 4, 5details
#4ACCEPTED0.01 s2, 4, 5details
#5ACCEPTED0.03 s2, 4, 5details
#6ACCEPTED0.02 s4, 5details
#7ACCEPTED0.03 s4, 5details
#8ACCEPTED0.03 s4, 5details
#9ACCEPTED0.03 s1, 3, 4, 5details
#10ACCEPTED0.01 s1, 3, 4, 5details
#11ACCEPTED0.03 s1, 3, 4, 5details
#12ACCEPTED0.02 s1, 3, 4, 5details
#130.03 s1, 3, 4, 5details
#140.02 s1, 3, 4, 5details
#15ACCEPTED0.03 s1, 3, 4, 5details
#16ACCEPTED0.02 s1, 3, 4, 5details
#17ACCEPTED0.03 s1, 2, 3, 4, 5details
#180.02 s1, 3, 4, 5details
#19ACCEPTED0.09 s2, 5details
#20ACCEPTED0.07 s2, 5details
#21ACCEPTED0.08 s2, 5details
#22ACCEPTED0.07 s5details
#230.07 s5details
#240.02 s3, 4, 5details
#250.02 s3, 4, 5details
#260.02 s3, 4, 5details
#270.03 s3, 4, 5details
#280.02 s4, 5details
#290.03 s4, 5details
#300.03 s4, 5details
#310.04 s4, 5details
#320.02 s2, 4, 5details
#330.03 s2, 4, 5details
#340.03 s2, 4, 5details
#350.02 s2, 4, 5details
#360.07 s3, 5details
#370.06 s3, 5details
#380.06 s3, 5details
#390.06 s3, 5details
#400.07 s5details
#410.06 s5details
#420.07 s5details
#430.06 s5details
#440.06 s2, 5details
#450.06 s2, 5details
#460.07 s2, 5details
#470.07 s2, 5details

Code

#include <bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define ld long double
#define N (1<<18)
#define M 1000000007
#define P complex<long long>
#define X real()
#define Y imag()

using namespace std;

int n, k, m, ep[101010], viimVankE[101010], vangittu[101010], vankilaNro[101010];
set<pair<int,int>> vank[101010];
set<int> tVank; // Tyhjä vankila vapautusaika_vankilaId
priority_queue<int> vva[101010]; // Vankilan vapautusajat negatiivisina
set<int> vangitut, tyhjatV;
char et[101010];

void ei() {
	cout << "IMPOSSIBLE";
	exit(0);
}

int main() {
	cin.tie(0);
	cout.tie(0);
	ios_base::sync_with_stdio(0);
	cin >> n >> k >> m;
	for(int i=1; i<=m; i++) {
		cin >> et[i];
		cin >> ep[i];
	}

	//Numeroi vankilat
	int cnt = 1;
	for(int i=1; i<=m; i++) {
		if(et[i] == 'O') {
			vankilaNro[i] = cnt;
			cnt++;
			if(cnt>k)
				cnt = 1;
		}
	}
	// Vangit vankiloihin
	int viimVapVank = 0;
	for(int i=m; i>=1; i--) {
		if(et[i] == 'O') {
			viimVapVank = i; // viimeisin vapaa vankila
			viimVankE[ep[i]] = i-1; // Tämän vangin oltava tällöin vapaa että voi avata
		} else { // napataan vanki
			if(!viimVapVank && viimVankE[ep[i]]) // ei vankilaa josta vapautetan
				ei();
			if(viimVapVank) {
				if(viimVankE[ep[i]] && viimVapVank > viimVankE[ep[i]]) // huom ei >=
					ei();
				if(viimVankE[ep[i]]) { // pitää vapautua x mennessä
					vank[viimVapVank].insert(make_pair(viimVankE[ep[i]], ep[i])); // oltava vapaa x, vanki y
				} else { // ei rajaa
					vank[viimVapVank].insert(make_pair(1e9, ep[i])); // oltava vapaa x, vanki y
				}
			}
			viimVankE[ep[i]] = i-1; // Tämän vangin oltava tällöin vapaa(että voidaan napata)
		}
	}
	// Siirto seuraaviin
	for(int i=m; i>=1; i--) {
		if(et[i] == 'O') {
			if(vank[i].size() == 0) { // Vankila tyhjä
				if(tyhjatV.find(vankilaNro[i]) != tyhjatV.end()) {
					// Edellinen tyhjä ei ehtinyt täyttyä
					ei();
				}
				tVank.insert(i);
				tyhjatV.insert(vankilaNro[i]);
			} else { // vankilassa ehkä siirrettäviä vankeja, siirretään vankeja tyhjiin vankiloihin ensimmäisenä viimeiseen tyhjään 
				auto it = vank[i].end();
				it--;
				vector<pair<int,int>> poisTasta;
				while(it != vank[i].begin()) { // ei siirretä ekaa koskaan että joku jää
					auto it2 = tVank.upper_bound((*it).first);
					if(it2 != tVank.begin()) {
						it2--;
						// Siirretään it2 vankilaan
						vank[*it2].insert(*it);
						poisTasta.push_back(*it);
						tyhjatV.erase(vankilaNro[*it2]);
						tVank.erase(it2);
					}
					it--;	
				}
				for(auto u:poisTasta)
					vank[i].erase(u);
			
			}
		}
	}
	// Testataan onko validi ja muodostetaan vastaus
	// 1. vapautuvat vankilat ei tyhjiä
	vector<int> ans;
	for(int i=m; i>=1; i--) {
		if(et[i] == 'O') {
			if(vank[i].size() == 0)
				ei();
			for(auto u:vank[i]) {
				vangittu[u.second] = vankilaNro[i];
			}
			ans.push_back(vankilaNro[i]);
		} else {
			ans.push_back(vangittu[i]);
		}
	}
	reverse(ans.begin(), ans.end());
	for(auto u:ans) {
		if(u)
		cout << u << " ";
		else
			cout << 1 << " ";
	}
}

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
1 1 1 2 

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
1 1 1 2 

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:

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

correct output
1 1 1 1 1 

user output
2 1 1 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 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

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 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

Test 23

Group: 5

Verdict:

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

correct output
IMPOSSIBLE

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

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 1 2 1 1 1 2 1 1 2 2 1 1 1 ...

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 1 2 1 1 2 2 1 1 2 2 1 1 1 ...

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 1 2 1 1 1 1 1 2 2 1 1 1 2 ...

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 1 2 2 1 1 1 2 2 1 1 1 1 1 ...

Test 28

Group: 4, 5

Verdict:

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
71 1 1 2 1 3 140 4 1 5 1 6 1 7...

Test 29

Group: 4, 5

Verdict:

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 1 2 1 3 171 4 1 5 1 6 1 1 ...

Test 30

Group: 4, 5

Verdict:

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 57 2 1 1 58 3 69 4 1 5 39 ...

Test 31

Group: 4, 5

Verdict:

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
7 1 9 1 2 1 82 95 77 2 64 105 ...

Test 32

Group: 2, 4, 5

Verdict:

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
71 1 1 2 1 3 140 4 1 5 1 6 1 7...

Test 33

Group: 2, 4, 5

Verdict:

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 1 2 1 3 171 4 1 5 1 6 1 1 ...

Test 34

Group: 2, 4, 5

Verdict:

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
106 1 57 2 1 141 58 3 129 4 15...

Test 35

Group: 2, 4, 5

Verdict:

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
6 1 7 27 2 1 1 1 61 2 32 27 1 ...

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 1 1 1 2 2 1 1 2 2 1 1 ...

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 1 2 1 1 2 2 2 1 1 2 1 1 1 ...

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 1 2 2 1 1 1 2 2 1 1 1 1 1 ...

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
2 1 1 1 1 1 1 2 1 1 1 2 1 2 1 ...

Test 40

Group: 5

Verdict:

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 40587 2 1493 3 39440 4 1 5...

Test 41

Group: 5

Verdict:

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 1 2 1 3 39442 4 1 5 10292 ...

Test 42

Group: 5

Verdict:

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
16298 1 1 2 34497 23779 36854 ...

Test 43

Group: 5

Verdict:

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
227 1 1 5139 1 6971 8325 9710 ...

Test 44

Group: 2, 5

Verdict:

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 40587 2 1493 3 39440 4 1 5...

Test 45

Group: 2, 5

Verdict:

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 1 2 1 3 39442 4 21511 5 10...

Test 46

Group: 2, 5

Verdict:

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 1 2 1 10664 31787 3 1 4 57...

Test 47

Group: 2, 5

Verdict:

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
227 1 9125 18268 1 17837 19371...