CSES - Datatähti Open 2017 - Results
Submission details
Task:Mex value
Sender:fjuengermann
Submission time:2017-01-22 12:43:30 +0200
Language:Java
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.18 s1details
#20.15 s1details
#30.22 s1details
#40.16 s1details
#5ACCEPTED0.16 s1details
#60.19 s1details
#70.18 s1details
#80.17 s1details
#90.17 s1details
#10ACCEPTED0.17 s1details
#11--2details
#12--2details
#13--2details
#14--2details
#15ACCEPTED0.95 s2details
#16--2details
#17--2details
#18--2details
#19--2details
#20ACCEPTED0.94 s2details

Code

import java.util.*;
/**
* Created by Florian on 22-Jan-17.
*/
public class MexValue2 {
public static void main(String args[]) {
Scanner sc = new Scanner(System.in);
StringTokenizer st = new StringTokenizer(sc.nextLine());
int n = Integer.parseInt(st.nextToken());
int k = Integer.parseInt(st.nextToken());
st = new StringTokenizer(sc.nextLine());
int[] nums = new int[n];
for (int i = 0; i < n; i++) {
nums[i] = Integer.parseInt(st.nextToken());
}
TreeMap<Integer, Integer> treeMap = new TreeMap<>();
PriorityQueue<Integer> holes = new PriorityQueue<>();
for (int i = 0; i < k; i++) {
int count = 1;
if (treeMap.containsKey(nums[i])) {
count = treeMap.get(nums[i]);
}
treeMap.put(nums[i], count);
}
int last = -1;
for (Integer key : treeMap.keySet()) {
if (key - last > 1) {
//hole izz da
holes.add(key);
}
last = key;
}
int posL = 0, posH = k;
while (posH <= n) {
if (holes.size() > 0) {
int hole = holes.peek();
Integer lower = treeMap.lowerKey(hole);
if (lower == null)
lower = -1;
System.out.print((lower + 1));
} else {
System.out.print((treeMap.lastKey() + 1));
}
if (posH < n) {
int rm = nums[posL];
int add = nums[posH];
int count = treeMap.get(rm);
if (count == 1) {
treeMap.remove(rm);
Integer higher = treeMap.higherKey(rm);
if (higher != null)
holes.add(higher);
} else {
treeMap.put(rm, count - 1);
}
//add
if (treeMap.containsKey(add)) {
treeMap.put(add, treeMap.get(add) + 1);
} else {
treeMap.put(add, 1);
Integer next = treeMap.higherKey(add);
if (next != null) {
if (next - add == 1 && holes.contains(next)) {
Integer lower = treeMap.lowerKey(add);
if (lower == null)
lower = -1;
if (add - lower == 1) {
holes.remove(next);
}
}
}
}
System.out.print(' ');
}
posL++;
posH++;
}
System.out.println();
}
}

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
4 10 9 1 1 1 1 1 1 1 1 2 2 2 0...

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

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
7 10 9 2 2 1 0 1 1 1 2 3 3 3 3...

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
9 10 3 0 0 0 0 0 4 2 2 2 4 4 4...

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
8 10 0 5 1 1 1 1 1 5 2 5 7 0 0...

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
6 10 1 1 1 1 2 0 0 2 2 2 2 2 2...

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
2 10 5 0 0 0 0 0 0 0 0 1 1 1 1...

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
8 10 7 3 0 0 0 0 7 5 5 5 7 7 1...

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