CSES - Datatähti 2017 loppu - Results
Submission details
Task:Mex-luku
Sender:mangolassi
Submission time:2017-01-19 12:47:02 +0200
Language:C++
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.18 s1details
#20.21 s1details
#30.18 s1details
#40.21 s1details
#5ACCEPTED0.04 s1details
#60.21 s1details
#70.18 s1details
#80.20 s1details
#90.17 s1details
#10ACCEPTED0.05 s1details
#110.20 s2details
#120.21 s2details
#130.18 s2details
#140.20 s2details
#15ACCEPTED0.09 s2details
#160.20 s2details
#170.22 s2details
#180.21 s2details
#190.18 s2details
#20ACCEPTED0.07 s2details

Code

#include <iostream>

struct BIT {
	int* sum;
	int n;

	BIT(int size) {
		n = size;
		sum = new int[n+1];
		for (int i = 1; i < n+1; ++i) {
			sum[i] = 0;
		}
	}

	void change(int i, int c) {
		++i;
		while(i <= n) {
			sum[i] += c;
			i += (i & -i);
		}
	}

	int get(int i) {
		++i;
		int res = 0;
		while(i > 0) {
			res += sum[i];
			i -= (i & -i);
		}
		return res;
	}
		
	void print() {
		for (int i = 1; i < n+1; ++i) {
			std::cout << sum[i] << " ";
		}
		std::cout << "\n";
	}
};

const int N = 1e5;
int amount [N];
int nums   [N];

int main() {
	std::ios_base::sync_with_stdio(false);
	std::cin.tie(0);

	int n, k;
	std::cin >> n >> k;
	for (int i = 0; i < n; ++i) {
		std::cin >> nums[i];
	}
	for (int i = 0; i < k + 1; ++i) {
		amount[i] = 0;
	}
	BIT tree (k + 1);
	for (int i = 0; i < k; ++i) {
		if (! amount[nums[i]]) {
			tree.change(nums[i], 1);
		}
		++amount[nums[i]];
	}
	for (int i = 0; i + k < n; ++i) {
		//tree.print();
		int low = 0;
		int high = k+1;
		while(low != high) {
			int mid = (low + high) / 2;
			if (tree.get(mid) > mid) {
				low = mid + 1;
			} else {
				high = mid;
			}
		}
		std::cout << low << " ";

		int leaver = nums[i];
		int joiner = nums[i + k];
		--(amount[leaver]);
		if (amount[leaver] == 0) {
			tree.change(leaver, -1);
		}
		++(amount[joiner]);
		if (amount[joiner] == 1) {
			tree.change(joiner, 1);
		}
	}
	int low = 0;
	int high = k+1;
	while(low != high) {
		int mid = (low + high) / 2;
		if (tree.get(mid) > mid) {
			low = mid + 1;
		} else {
			high = mid;
		}
	}
	std::cout << low << "\n";
}

Test details

Test 1

Group: 1

Verdict:

input
100 10
1000000000 9 1 0 5 7 2 8 6 3 4...

correct output
4 10 9 1 1 1 1 1 1 1 1 4 10 3 ...

user output
(empty)

Test 2

Group: 1

Verdict:

input
100 10
1000000000 4 3 8 1 0 9 5 7 2 6...

correct output
6 10 4 3 3 1 1 1 1 1 3 3 10 6 ...

user output
(empty)

Test 3

Group: 1

Verdict:

input
100 10
1000000000 9 2 3 1 0 8 5 6 4 7...

correct output
7 10 9 2 2 1 0 1 1 1 2 9 10 4 ...

user output
(empty)

Test 4

Group: 1

Verdict:

input
100 10
1000000000 3 0 4 6 5 7 1 2 8 9...

correct output
9 10 3 0 0 0 0 0 4 2 2 2 10 6 ...

user output
(empty)

Test 5

Group: 1

Verdict: ACCEPTED

input
100 10
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 ...

correct output
10 10 10 10 10 10 10 10 10 10 ...

user output
10 10 10 10 10 10 10 10 10 10 ...

Test 6

Group: 1

Verdict:

input
100 10
1000000000 0 5 1 9 3 4 6 7 2 8...

correct output
8 10 0 5 1 1 1 1 1 5 2 5 10 0 ...

user output
(empty)

Test 7

Group: 1

Verdict:

input
100 10
1000000000 1 2 9 3 8 0 4 7 5 6...

correct output
6 10 1 1 1 1 2 0 0 2 3 9 10 6 ...

user output
(empty)

Test 8

Group: 1

Verdict:

input
100 10
1000000000 5 0 6 7 1 4 8 3 9 2...

correct output
2 10 5 0 0 0 0 0 0 0 0 1 10 9 ...

user output
(empty)

Test 9

Group: 1

Verdict:

input
100 10
1000000000 7 3 0 1 4 9 2 5 6 8...

correct output
8 10 7 3 0 0 0 0 7 5 5 5 10 8 ...

user output
(empty)

Test 10

Group: 1

Verdict: ACCEPTED

input
100 10
0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 ...

correct output
10 10 10 10 10 10 10 10 10 10 ...

user output
10 10 10 10 10 10 10 10 10 10 ...

Test 11

Group: 2

Verdict:

input
100000 10000
1000000000 4786 3512 3285 1919...

correct output
9547 10000 4786 3512 3285 1919...

user output
(empty)

Test 12

Group: 2

Verdict:

input
100000 10000
1000000000 7981 3955 790 45 34...

correct output
7657 10000 7981 3955 790 45 45...

user output
(empty)

Test 13

Group: 2

Verdict:

input
100000 10000
1000000000 2329 1825 9435 3800...

correct output
5701 10000 2329 1825 1825 1825...

user output
(empty)

Test 14

Group: 2

Verdict:

input
100000 10000
1000000000 2754 6029 8007 6286...

correct output
1423 10000 2754 2754 2754 2754...

user output
(empty)

Test 15

Group: 2

Verdict: ACCEPTED

input
100000 10000
0 1 2 3 4 5 6 7 8 9 10 11 12 1...

correct output
10000 10000 10000 10000 10000 ...

user output
10000 10000 10000 10000 10000 ...

Test 16

Group: 2

Verdict:

input
100000 10000
1000000000 7560 4634 7044 3853...

correct output
9855 10000 7560 4634 4634 3853...

user output
(empty)

Test 17

Group: 2

Verdict:

input
100000 10000
1000000000 7945 6674 3975 3002...

correct output
6373 10000 7945 6674 3975 3002...

user output
(empty)

Test 18

Group: 2

Verdict:

input
100000 10000
1000000000 2506 6827 6871 8593...

correct output
2978 10000 2506 2506 2506 2506...

user output
(empty)

Test 19

Group: 2

Verdict:

input
100000 10000
1000000000 2769 5036 539 4409 ...

correct output
2249 10000 2769 2769 539 539 5...

user output
(empty)

Test 20

Group: 2

Verdict: ACCEPTED

input
100000 10000
0 1 2 3 4 5 6 7 8 9 10 11 12 1...

correct output
10000 10000 10000 10000 10000 ...

user output
10000 10000 10000 10000 10000 ...