CSES - NOI 2019 - Results
Submission details
Task:Thieves and Prisons
Sender:Olli Järviniemi
Submission time:2019-03-06 13:51:28 +0200
Language:C++
Status:READY
Result:8
Feedback
groupverdictscore
#1ACCEPTED8
#20
#30
#40
#50
Test results
testverdicttimegroup
#1ACCEPTED0.03 s2, 4, 5details
#2ACCEPTED0.02 s2, 4, 5details
#3ACCEPTED0.02 s2, 4, 5details
#4ACCEPTED0.03 s2, 4, 5details
#5ACCEPTED0.04 s2, 4, 5details
#6ACCEPTED0.03 s4, 5details
#7ACCEPTED0.03 s4, 5details
#8ACCEPTED0.03 s4, 5details
#9ACCEPTED0.02 s1, 3, 4, 5details
#10ACCEPTED0.03 s1, 3, 4, 5details
#11ACCEPTED0.03 s1, 3, 4, 5details
#12ACCEPTED0.02 s1, 3, 4, 5details
#13ACCEPTED0.03 s1, 3, 4, 5details
#14ACCEPTED0.04 s1, 3, 4, 5details
#15ACCEPTED0.03 s1, 3, 4, 5details
#16ACCEPTED0.03 s1, 3, 4, 5details
#17ACCEPTED0.03 s1, 2, 3, 4, 5details
#18ACCEPTED0.03 s1, 3, 4, 5details
#19ACCEPTED0.17 s2, 5details
#20ACCEPTED0.19 s2, 5details
#21ACCEPTED0.20 s2, 5details
#22ACCEPTED0.17 s5details
#23ACCEPTED0.15 s5details
#240.03 s3, 4, 5details
#250.03 s3, 4, 5details
#260.03 s3, 4, 5details
#270.03 s3, 4, 5details
#280.03 s4, 5details
#290.02 s4, 5details
#300.04 s4, 5details
#310.04 s4, 5details
#320.03 s2, 4, 5details
#330.03 s2, 4, 5details
#340.04 s2, 4, 5details
#350.03 s2, 4, 5details
#360.10 s3, 5details
#370.10 s3, 5details
#380.10 s3, 5details
#390.11 s3, 5details
#400.11 s5details
#410.10 s5details
#420.10 s5details
#430.10 s5details
#440.09 s2, 5details
#450.10 s2, 5details
#460.11 s2, 5details
#470.10 s2, 5details

Compiler report

input/code.cpp: In function 'int main()':
input/code.cpp:129:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 0; j < del[i].size(); ++j) {
                  ~~^~~~~~~~~~~~~~~
input/code.cpp:157:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < event.size(); ++i) {
                 ~~^~~~~~~~~~~~~~
input/code.cpp:175:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 0; j < del[i].size(); ++j) {
                  ~~^~~~~~~~~~~~~~~
input/code.cpp:217:21: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(int j = 0; j < cont[ans[i-1]].size(); ++j) {
                   ~~^~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:230:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < ans.size(); ++i) {
                 ~~^~~~~~~~~~~~
input/code.c...

Code

#include <iostream>
#include <set>
#include <vector>
#include <algorithm>

using namespace std;

const int N = 1e5 + 5;

typedef pair<int, int> pii;
typedef long long ll;

set<int> per[N];

int pri[N];
int mi[N];
int w[N];

vector<int> del[N];
set<int> fre;


vector<int> loc[N];

bool impri[N];
vector<int> cont[N];

int fi[N];

int label[N];

int main() {
	int n, k, m;
	cin >> n >> k >> m;
	vector<int> g;
	vector<char> t;
	g.push_back(-1);
	t.push_back(-1);
	for(int i = 1; i <= m; ++i) {
		char a; int b;
		cin >> a >> b;
		t.push_back(a);
		g.push_back(b);
		per[b].insert(i);
		if(b > n) {
			cout << "IMPOSSIBLE\n";
			return 0;
		}
	}


	if(t[1] == 'C') {
		pri[1] = 1;
	} else {
		cout << "IMPOSSIBLE\n";
		return 0;
	}

	for(int i = 2; i <= m; ++i) {
		if(t[i] == 'C') {
			pri[i] = pri[i-1] + 1;
		} else {
			pri[i] = pri[i-1] - 1;
		}
	}
	mi[m] = pri[m];
	for(int i = m-1; i >= 1; --i) {
		mi[i] = min(mi[i+1], pri[i]);
	}

	if(mi[1] < 0) {
		cout << "IMPOSSIBLE\n";
		return 0;
	}

	int ext = 0;
	int curPri = 0;
	for(int i = 1; i <= m; ++i) {
		if(t[i] == 'C') {
			++curPri;
			continue;
		}
		w[i] = min(mi[i] + 1 - ext, curPri);
		curPri -= w[i];
		ext += w[i] - 1;

	}

	set<pair<int, int> > ord;
	set<int> people;

	for(int i = 1; i <= m; ++i) {
		if(t[i] == 'C') {
			if(people.count(g[i]) > 0) {
				cout << "IMPOSSIBLE\n";
				return 0;
			}
			people.insert(g[i]);
			set<int>::iterator it = per[g[i]].upper_bound(i);
			if(it == per[g[i]].end()) {
				ord.insert({N, i});
			} else {
				ord.insert({*(per[g[i]].upper_bound(i)), i});
//				ord.insert({g[i], g[i]});
			}
			continue;
		}
		for(int j = 1; j <= w[i]; ++j) {
			auto it = ord.begin();
			int abb = (*it).second;
			del[i].push_back(abb);
	//		cout << "Adding " << abb << "\n";
			people.erase((*it).second);
			ord.erase(it);

		}
	}

	for(int i = 1; i <= k; ++i) {
		fre.insert(i);
	}

	vector<pair<int, int> > event;

	for(int i = 1; i <= m; ++i) {
		if(t[i] == 'C') continue;

		fi[i] = N;
		for(int j = 0; j < del[i].size(); ++j) {
			int mm = del[i][j];
			fi[i] = min(fi[i], mm);
		}
		event.push_back({fi[i], i});
		event.push_back({i, i});
	}

	sort(event.begin(), event.end());

/*	for(int i = 0; i < event.size(); ++i) {
		cout << event[i].first << " " << event[i].second << "\n";
	}

	for(int i = 1; i <= m; ++i) {
		cout << w[i] << " ";
	}
	cout << "\n";
	for(int i = 1; i <= m; ++i) {
		cout << "Opening: ";
		for(int j = 0; j < del[i].size(); ++j) {
			cout << del[i][j] << " ";
		}
		cout << "\n";
	}

	cout << "\n\n";
*/
	for(int i = 0; i < event.size(); ++i) {
		int a = event[i].first;
		int b = event[i].second;
		if(a == b) {
			fre.insert(label[a]);
			continue;
		}
		if(fre.size() == 0) {
			cout << "IMPOSSIBLE\n";
			return 0;
		}
		int lab = *(fre.begin());
		label[b] = lab;
		fre.erase(fre.begin());
	}

	for(int i = 1; i <= m; ++i) {
		if(t[i] == 'C') continue;
		for(int j = 0; j < del[i].size(); ++j) {
			int mm = del[i][j];
			loc[g[mm]].push_back(label[i]);
		}
		loc[g[i]].push_back(label[i]);
	}


	for(int i = 1; i <= m; ++i) {
		reverse(loc[i].begin(), loc[i].end());
	}

/*	for(int i = 1; i <= n; ++i) {
		cout << "The prisons for " << i << ": ";
		for(int j = 0; j < loc[i].size(); ++j) {
			cout << loc[i][j] << " ";
		}
		cout << "\n";
	}
*/

	vector<int> ans;
	for(int i = 1; i <= m; ++i) {
		if(loc[g[i]].size() == 0) {
			ans.push_back(1);
		} else {
			ans.push_back(loc[g[i]][loc[g[i]].size() - 1]);
			loc[g[i]].pop_back();
		}

		if(t[i] == 'C') {
			cont[ans[i-1]].push_back(g[i]);
			if(impri[g[i]]) {
				cout << "IMPOSSIBLE\n";
				return 0;
			}
			impri[g[i]] = true;
		} else {
			if(impri[g[i]]) {
				cout << "IMPOSSIBLE\n";
				return 0;
			}
			for(int j = 0; j < cont[ans[i-1]].size(); ++j) {
				int abb = cont[ans[i-1]][j];
				if(!impri[abb]) {
					cout << "IMPOSSIBLE\n";
					return 0;
				}
				impri[abb] = false;
			}
			cont[ans[i-1]].clear();

		}
	}

	for(int i = 0; i < ans.size(); ++i) {
		cout << ans[i] << " ";
	}

	cout << "\n";
	return 0;

	cout << "\n\n";
	for(int i = 1; i <= m; ++i) {
		cout << del[i].size() << ": ";
		for(int j = 0; j < del[i].size(); ++j) {
			cout << del[i][j] << " ";
		}
		cout << "\n";
	}



}

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: ACCEPTED

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

correct output
1 1 1 1 

user output
1 1 1 1 

Test 14

Group: 1, 3, 4, 5

Verdict: ACCEPTED

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

correct output
1 1 1 1 

user output
1 1 1 1 

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

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

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

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

Test 23

Group: 5

Verdict: ACCEPTED

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

correct output
IMPOSSIBLE

user output
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE

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
IMPOSSIBLE