CSES - NOI 2019 - Results
Submission details
Task:Distance Code
Sender:William Bille Meyling
Submission time:2019-03-06 15:53:43 +0200
Language:Java
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#1ACCEPTED0.26 s1, 2, 3details
#2ACCEPTED0.26 s1, 2, 3details
#3ACCEPTED0.24 s1, 2, 3details
#4ACCEPTED0.25 s1, 2, 3details
#50.25 s1, 2, 3details
#6ACCEPTED0.25 s1, 2, 3details
#7ACCEPTED0.25 s1, 2, 3details
#8ACCEPTED0.26 s1, 2, 3details
#90.24 s1, 2, 3details
#100.25 s1, 2, 3details
#110.24 s1, 2, 3details
#12ACCEPTED0.29 s2, 3details
#130.27 s2, 3details
#140.28 s2, 3details
#150.29 s2, 3details
#16--3details
#17--3details
#18--3details
#19--3details
#20ACCEPTED0.24 s1, 2, 3details

Code

//thinking depth first with iterative deepening maybe (though slow)
//using heap

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.HashSet;
import java.util.PriorityQueue;

public class Distance_Code_v3 {

    private static class Node implements Comparable<Node>{

        int number;
        HashSet<Node> nbs = new HashSet<>();

        public Node(int number){
            this.number = number;
        }

        @Override
        public int compareTo(Node o) {
            return Integer.compare(this.nbs.size(), o.nbs.size());
        }
    }

    private static class Node2{

        int number;
        ArrayList<Node2> children = new ArrayList<>();

        public Node2(int number){
            this.number = number;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int type = Integer.valueOf(br.readLine());
        int N = Integer.valueOf(br.readLine());
        if(type == 1){
            ArrayList<Node> nodes = new ArrayList<>();
            for(int i = 0; i < N; i++) nodes.add(null);
            for(int i = 0; i < N - 1; i++){
                String[] input = br.readLine().split(" ");
                int a = Integer.valueOf(input[0])-1;
                int b = Integer.valueOf(input[1])-1;
                if(nodes.get(a) == null){
                    nodes.set(a, new Node(a));
                }
                if(nodes.get(b) == null) {
                    nodes.set(b, new Node(b));
                }
                nodes.get(a).nbs.add(nodes.get(b));
                nodes.get(b).nbs.add(nodes.get(a));
            }
            nodes.sort(new Comparator<Node>() {
                @Override
                public int compare(Node o1, Node o2) {
                    return Integer.compare(o1.nbs.size(), o2.nbs.size());
                }
            });
            PriorityQueue<Node> heap = new PriorityQueue<>(nodes);
            ArrayList<Integer> result = new ArrayList<>();
            while(!heap.isEmpty()){
                Node curr = heap.poll();
                result.add(curr.number);
                curr.nbs.remove(curr);
            }
            for(int i = 0; i < result.size(); i++){
                System.out.print((result.get(i)+1) + " ");
            }
        }else{
            inputs = br.readLine().split(" ");
            //if distance == 2 -> same depth && same parent
            //if distance == 1 -> its the parent
            //if distance == 3 -> different depth "different parent"
            //if distance == 4 -> same depth different parent
            //no its wrong
            inputIndex = inputs.length-1;
            Node2 root = new Node2(nextNumber);
            ArrayList<Node2> temp = new ArrayList<>();
            temp.add(root);
            decode(temp, 1);
            dfs(root);
        }
    }

    private static int nextNumber = 1;
    private static int inputIndex;
    private static String[] inputs;

    private static void decode(ArrayList<Node2> siblings, int depth){
        int siblingIndex = 0;
        while(inputIndex > -1){
            int diff = Integer.valueOf(inputs[inputIndex]);
            inputIndex--;
            if(diff == 1 || diff == 2 || diff == depth+1){
                nextNumber++;
                Node2 child = new Node2(nextNumber);
                siblings.get(siblingIndex).children.add(child);
            }else if(diff == 2*depth){ //assuming a depth shift is never on the curr sibling
                siblingIndex++;
                nextNumber++;
                Node2 child = new Node2(nextNumber);
                siblings.get(siblingIndex).children.add(child);
            }else{
                inputIndex++;
                break;
            }
        }
        for(int i = 0; i < siblings.size(); i++){
            if(siblings.get(i).children.size() != 0){
                decode(siblings.get(i).children, depth + 1);
            }
        }
    }

    private static void dfs(Node2 node){
        for(int i = 0; i < node.children.size(); i++){
            System.out.println(node.number + " " + node.children.get(i).number);
            dfs(node.children.get(i));
        }
    }
}

Test details

Test 1

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
2
2 1

correct output
(empty)

user output
1 2

Test 2

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
3
3 1
2 1

correct output
(empty)

user output
1 2
1 3

Test 3

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
4
3 2
2 1
4 1

correct output
(empty)

user output
1 2
2 4
1 3

Test 4

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
4
2 3
3 4
1 3

correct output
(empty)

user output
1 2
1 3
1 4

Test 5

Group: 1, 2, 3

Verdict:

input
1
5
3 5
4 1
1 3
...

correct output
(empty)

user output
1 2
1 3
3 5
1 4

Test 6

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
5
3 2
3 4
5 1
...

correct output
(empty)

user output
1 2
2 4
2 5
1 3

Test 7

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
5
4 3
1 4
4 2
...

correct output
(empty)

user output
1 2
1 3
1 4
1 5

Test 8

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
10
9 3
8 9
2 9
...

correct output
(empty)

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

Test 9

Group: 1, 2, 3

Verdict:

input
1
10
9 2
5 8
7 1
...

correct output
(empty)

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

Test 10

Group: 1, 2, 3

Verdict:

input
1
10
10 4
9 1
4 7
...

correct output
(empty)

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

Test 11

Group: 1, 2, 3

Verdict:

input
1
10
2 6
4 3
3 5
...

correct output
(empty)

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

Test 12

Group: 2, 3

Verdict: ACCEPTED

input
1
500
10 6
6 255
6 428
...

correct output
(empty)

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

Test 13

Group: 2, 3

Verdict:

input
1
500
152 466
451 313
158 479
...

correct output
(empty)

user output
72 106 499 498 497 496 495 494...

Test 14

Group: 2, 3

Verdict:

input
1
500
109 440
330 190
443 161
...

correct output
(empty)

user output
2 3 6 12 27 57 115 239 241 116...

Test 15

Group: 2, 3

Verdict:

input
1
500
144 373
257 233
341 318
...

correct output
(empty)

user output
25 28 39 111 159 343 346 167 3...

Test 16

Group: 3

Verdict:

input
1
100000
54983 75172
93807 75172
44082 75172
...

correct output
(empty)

user output
(empty)

Test 17

Group: 3

Verdict:

input
1
100000
88863 19059
86423 76688
98536 95984
...

correct output
(empty)

user output
(empty)

Test 18

Group: 3

Verdict:

input
1
100000
59979 6389
19097 24999
27846 82330
...

correct output
(empty)

user output
(empty)

Test 19

Group: 3

Verdict:

input
1
100000
58761 66001
25102 51081
98625 67861
...

correct output
(empty)

user output
(empty)

Test 20

Group: 1, 2, 3

Verdict: ACCEPTED

input
1
6
2 1
3 2
4 2
...

correct output
(empty)

user output
1 2
2 5
2 6
1 3
1 4