CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:iRobo
Submission time:2020-10-11 23:58:19 +0300
Language:Java
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#10.14 s1, 2details
#20.18 s1, 2details
#30.18 s1, 2details
#40.17 s1, 2details
#50.17 s1, 2details
#60.13 s1, 2details
#70.18 s1, 2details
#80.18 s1, 2details
#90.13 s1, 2details
#100.17 s1, 2details
#110.14 s2details
#120.18 s2details
#130.18 s2details
#140.18 s2details
#150.18 s2details
#160.18 s2details
#170.18 s2details
#180.18 s2details
#190.18 s2details
#200.14 s2details
#210.20 s2details
#220.20 s2details
#230.20 s2details
#240.19 s2details
#250.20 s2details
#260.19 s2details
#270.19 s2details
#280.20 s2details
#290.19 s2details
#300.20 s2details
#31ACCEPTED0.14 s1, 2details

Code

import java.util.Arrays;
import java.util.Scanner;

public class Main {

    static int[] tempCubeArray = new int[6];

    public static void main(String[] args) {

        Scanner input = new Scanner(System.in);
        int cubeCount = input.nextInt();
        input.nextLine();

        int[][][] cubes = new int[cubeCount][5][5];
        int[][] cubeArray = new int[cubeCount][6];

        //get cubes from input into 2d arrays
        for(int i = 0; i < cubeCount; i++) {
            for(int j = 0; j < 5; j++) {
                String lineInput = input.nextLine();
                for(int k = 0; k < 5; k++) {
                    if(lineInput.charAt(k) == '.') {
                        cubes[i][j][k] = 0;
                    } else {
                        cubes[i][j][k] = Integer.parseInt(String.valueOf(lineInput.charAt(k)));
                    }
                }
            }
            input.nextLine();
        }

        for(int i = 0; i < cubeCount; i++) {
            for(int j = 0; j < 5; j++) {
                for(int k = 0; k < 5; k++) {
                    //System.out.print(cubes[i][j][k] + " ");
                }
                //System.out.println();
            }
            //System.out.println();
            //System.out.println();
        }

        for(int i = 0; i < cubeCount; i++) {
            int[] tempArray = convertCubeToArray(cubes[i]);
            for(int j = 0; j < 6; j++) {
                cubeArray[i][j] = tempArray[j];
            }
            if(cubeArray[i][3] == cubeArray[i][5]) { //weird bug fix?
                for(int k = 0; k < 6; k++) {
                    boolean check = true;
                    for(int m = 0; m < 6; m++) {
                        if(cubeArray[i][m] == k+1) {
                            check = false;
                        }
                    }
                    if(check) {
                        cubeArray[i][3] = k+1;
                    }
                }
            }
        }


        for(int i = 0; i < cubeCount; i++) {
            for(int j = 0; j < 6; j++) {
                //System.out.print(cubeArray[i][j] + " ");
            }
            //System.out.println(Arrays.toString(cubeArray[i]));
            //System.out.println();
        }

        //cubes then need to be rotated into the correct orientation! Smallest number up top

        for(int i = 0; i < cubeArray.length; i++) {
            int[] tempArray = rotateCube(cubeArray[i]);
            for(int j = 0; j < 6; j++) {
                cubeArray[i][j] = tempArray[j];
            }
        }

        sendOutput(cubeArray);

    }

    private static int[] rotateCube(int[] cubeArray) {
        int smallest = 100;

        for(int i = 0; i < 4; i++) {
            if(cubeArray[i+1] < smallest) {
                smallest = cubeArray[i+1];
            }
        }

        for(int i = 0; i < 4; i++) {
            
        }
        return cubeArray;
    }

    private static void sendOutput(int[][] cubeArray) {
        for(int i = 0; i < cubeArray.length; i++) {
            boolean test = true;
            for(int j = 0; j < cubeArray.length; j++) {
                if(Arrays.equals(cubeArray[i],cubeArray[j]) && i != j) {
                    System.out.print(j + " ");
                    test = false;
                }
            }
            if(test) {
                System.out.print("-");
            }
            System.out.println();
        }
    }

    private static int[] convertCubeToArray(int[][] cube) {
        int[] cubeArray = new int[6];

        //find position of 1
        int x = 0, y = 0;
        for(int i = 0; i < 5; i++) {
            for(int j = 0; j < 5; j++) {
                if(cube[i][j] == 1) {
                    y = i;
                    x = j;
                }
            }
        }

        cubeArray[0] = 1;
        int[] previousTransforms = new int[5];
        Arrays.fill(tempCubeArray, 0);
        recursiveSolve(cube, y,x,1, cubeArray, previousTransforms, 0);
        cubeArray = tempCubeArray;

        return cubeArray;
    }

    private static void recursiveSolve(int[][] cube, int y, int x, int previous, int[] cubeArray, int[] previousTransforms, int depth) {
        boolean lastSlot = true;
        //check if there is a number above
        if(y-1 >= 0 && cube[y-1][x] != 0 && cube[y-1][x] != previous) {
            //System.out.println("transformUp");
            lastSlot = false;

            //create copies
            int[] cubeArray2 = cubeArray;
            int[] previousTransforms2 = previousTransforms;
            int depth2 = depth;

            cubeArray2[1] = cube[y-1][x];
            cubeArray2 = transformUp(cubeArray2);
            previousTransforms2[depth2] = 1;
            depth2++;
            recursiveSolve(cube,y-1,x,cube[y][x],cubeArray2,previousTransforms2,depth2);
        }
        //check if there is a number to the right
        if(x+1 < 5 && cube[y][x+1] != 0 && cube[y][x+1] != previous) {
            //System.out.println("transformRight");
            lastSlot = false;

            //create copies
            int[] cubeArray2 = cubeArray;
            int[] previousTransforms2 = previousTransforms;
            int depth2 = depth;

            cubeArray2[2] = cube[y][x+1];
            cubeArray2 = transformRight(cubeArray2);
            previousTransforms2[depth2] = 2;
            depth2++;
            recursiveSolve(cube,y,x+1,cube[y][x],cubeArray2,previousTransforms2,depth2);
        }
        //check if there is a number below
        if(y+1 < 5 && cube[y+1][x] != 0 && cube[y+1][x] != previous) {
            //System.out.println("transformDown");
            lastSlot = false;

            //create copies
            int[] cubeArray2 = cubeArray;
            int[] previousTransforms2 = previousTransforms;
            int depth2 = depth;

            cubeArray2[3] = cube[y+1][x];
            cubeArray2 = transformDown(cubeArray2);
            previousTransforms2[depth2] = 3;
            depth2++;
            recursiveSolve(cube,y+1,x,cube[y][x],cubeArray2,previousTransforms2,depth2);
        }
        //check if there is a number to the left
        if(x-1 >= 0 && cube[y][x-1] != 0 && cube[y][x-1] != previous) {
            //System.out.println("transformLeft");
            lastSlot = false;

            //create copies
            int[] cubeArray2 = cubeArray;
            int[] previousTransforms2 = previousTransforms;
            int depth2 = depth;

            cubeArray2[4] = cube[y][x-1];
            cubeArray2 = transformLeft(cubeArray2);
            previousTransforms2[depth2] = 4;
            depth2++;
            recursiveSolve(cube,y,x-1,cube[y][x],cubeArray2,previousTransforms2,depth2);
        }

        if(lastSlot) {
            int[] cubeArray2 = cubeArray;
            //reverse transformations to reach original cube rotation
            for(int i = depth-1; i >= 0; i--) {
                //System.out.println("anti-transforming " + previousTransforms[i]);
                if(previousTransforms[i] == 1) {
                    cubeArray2 = transformDown(cubeArray2);
                } else if(previousTransforms[i] == 2) {
                    cubeArray2 = transformLeft(cubeArray2);
                } else if(previousTransforms[i] == 3) {
                    cubeArray2 = transformUp(cubeArray2);
                } else if(previousTransforms[i] == 4) {
                    cubeArray2 = transformRight(cubeArray2);
                } else {
                    //System.out.println("error");
                }
            }
            //update the tempCubeArray with the new number positions on the cube, excluding 0 which is missing info
            for(int i = 0; i < 6; i++) {
                if(cubeArray2[i] != 0) {
                    if(tempCubeArray[i] == 0) {
                        tempCubeArray[i] = cubeArray2[i];
                    }
                }
            }
        }

    }

    private static int[] transformUp(int[] cube) {
        int temp = cube[0];
        cube[0] = cube[1];
        cube[1] = cube[5];
        cube[5] = cube[3];
        cube[3] = temp;
        return cube;
    }

    private static int[] transformRight(int[] cube) {
        int temp = cube[0];
        cube[0] = cube[2];
        cube[2] = cube[5];
        cube[5] = cube[4];
        cube[4] = temp;
        return cube;
    }

    private static int[] transformDown(int[] cube) {
        int temp = cube[0];
        cube[0] = cube[3];
        cube[3] = cube[5];
        cube[5] = cube[1];
        cube[1] = temp;
        return cube;
    }

    private static int[] transformLeft(int[] cube) {
        int temp = cube[0];
        cube[0] = cube[4];
        cube[4] = cube[5];
        cube[5] = cube[2];
        cube[2] = temp;
        return cube;
    }
}

Test details

Test 1

Group: 1, 2

Verdict:

input
3
165..
.4...
.3...
.2...
...

correct output
3
-
1

user output
-
-
-

Test 2

Group: 1, 2

Verdict:

input
5
264..
.5...
.3...
.1...
...

correct output
3
4 5
1
2 5
2 4

user output

-

-
-

Test 3

Group: 1, 2

Verdict:

input
5
152..
.4...
.3...
.6...
...

correct output
3 5
4
1 5
2
1 3

user output

-
-
-

Test 4

Group: 1, 2

Verdict:

input
5
142..
.6...
.3...
.5...
...

correct output
4 5
3
2
1 5
1 4

user output

-
-

-

Test 5

Group: 1, 2

Verdict:

input
5
123..
.4...
.6...
.5...
...

correct output
3
4 5
1
2 5
2 4

user output
-

-

-

Test 6

Group: 1, 2

Verdict:

input
5
213..
.6...
.4...
.5...
...

correct output
4 5
3
2
1 5
1 4

user output
-
-
-
-
-

Test 7

Group: 1, 2

Verdict:

input
5
314..
.5...
.2...
.6...
...

correct output
3
4 5
1
2 5
2 4

user output

-



Test 8

Group: 1, 2

Verdict:

input
5
163..
.2...
.5...
.4...
...

correct output
4 5
3
2
1 5
1 4

user output

-
-

-

Test 9

Group: 1, 2

Verdict:

input
5
264..
.1...
.3...
.5...
...

correct output
2 3
1 3
1 2
5
4

user output
-
-
-
-
-

Test 10

Group: 1, 2

Verdict:

input
5
214..
.3...
.5...
.6...
...

correct output
5
3 4
2 4
2 3
1

user output
-
-


-

Test 11

Group: 2

Verdict:

input
10
.41..
.5...
.2...
36...
...

correct output
9
4 6 8
7 10
2 6 8
-
...

user output
-
-
-
-
-
...

Test 12

Group: 2

Verdict:

input
10
5....
1436.
.2...
.....
...

correct output
5 8
6 10
4 7 9
3 7 9
1 8
...

user output


-
-
-
...

Test 13

Group: 2

Verdict:

input
10
2....
41...
.63..
.5...
...

correct output
4 9 10
5 6 7 8
-
1 9 10
2 6 7 8
...

user output
-

-
-

...

Test 14

Group: 2

Verdict:

input
10
1....
634..
..52.
.....
...

correct output
2 3 4 5 6 9 10
1 3 4 5 6 9 10
1 2 4 5 6 9 10
1 2 3 5 6 9 10
1 2 3 4 6 9 10
...

user output
5 9 
-
3 8 
2 8 
-
...

Test 15

Group: 2

Verdict:

input
10
.2...
4516.
3....
.....
...

correct output
5 7 9 10
8
4 6
3 6
1 7 9 10
...

user output
-
-
-
-
-
...

Test 16

Group: 2

Verdict:

input
10
.56..
.2...
.4...
31...
...

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

user output
-
-
-
-
-
...

Test 17

Group: 2

Verdict:

input
10
..62.
.31..
45...
.....
...

correct output
2 3 4 8
1 3 4 8
1 2 4 8
1 2 3 8
6 7 9 10
...

user output
-
-
-

5 9 
...

Test 18

Group: 2

Verdict:

input
10
532..
.4...
.1...
.6...
...

correct output
3 8 9
5 6
1 8 9
7 10
2 6
...

user output
-
-
-
-
-
...

Test 19

Group: 2

Verdict:

input
10
.64..
.1...
.3...
52...
...

correct output
2 5 6 7 8 9
1 5 6 7 8 9
4 10
3 10
1 2 6 7 8 9
...

user output
7 8 
4 5 6 
-
-
1 5 6 
...

Test 20

Group: 2

Verdict:

input
10
.4...
326..
.1...
.5...
...

correct output
4 7 8
6 9 10
5
1 7 8
3
...

user output
-
-
-
-
-
...

Test 21

Group: 2

Verdict:

input
20
.6...
.4...
31...
.25..
...

correct output
3 7 11 16
6
1 7 11 16
5 19
4 19
...

user output
2 10 

0 10 
-
-
...

Test 22

Group: 2

Verdict:

input
20
3....
5614.
..2..
.....
...

correct output
7 10 11 17 20
12
4 9 13 15 18
3 9 13 15 18
8 14 16
...

user output
9 19 
-
-
-
13 
...

Test 23

Group: 2

Verdict:

input
20
42...
.316.
.5...
.....
...

correct output
5 12 13 15 18
16 20
6 8 14
9 19
1 12 13 15 18
...

user output
4 12 
-

8 18 
0 12 
...

Test 24

Group: 2

Verdict:

input
20
..5..
.623.
41...
.....
...

correct output
2 6 11 12 13
1 6 11 12 13
5 16 18
7 14
3 16 18
...

user output
-
5 10 
17 
-
15 
...

Test 25

Group: 2

Verdict:

input
20
.46..
53...
.1...
.2...
...

correct output
2 3 5 7 15 17 19
1 3 5 7 15 17 19
1 2 5 7 15 17 19
8 10 11 14
1 2 3 7 15 17 19
...

user output
18 
4 6 14 
16 

1 6 14 
...

Test 26

Group: 2

Verdict:

input
20
.61..
.4...
35...
.2...
...

correct output
8 10 20
3 17 18 19
2 17 18 19
14 15
6 7 9 13
...

user output
-
17 
-
-
-
...

Test 27

Group: 2

Verdict:

input
20
..2..
1463.
.5...
.....
...

correct output
2 3 5 6 9 20
1 3 5 6 9 20
1 2 5 6 9 20
11 19
1 2 3 6 9 20
...

user output
1 8 19 
0 8 19 
-
18 

...

Test 28

Group: 2

Verdict:

input
20
...4.
5132.
6....
.....
...

correct output
2 8 10 12 13 19
1 8 10 12 13 19
4 5 15 16 17
3 5 15 16 17
3 4 15 16 17
...

user output
1 18 
0 18 

-

...

Test 29

Group: 2

Verdict:

input
20
.2...
.31..
45...
6....
...

correct output
5 8 9 14 17
3 10 16
2 10 16
13 15 19
1 8 9 14 17
...

user output
-


-
13 16 
...

Test 30

Group: 2

Verdict:

input
20
3....
452..
.1...
.6...
...

correct output
3 7 8 9 14 15 16 19
4 12 13 17
1 7 8 9 14 15 16 19
2 12 13 17
11 20
...

user output
14 15 18 

8 13 

19 
...

Test 31

Group: 1, 2

Verdict: ACCEPTED

input
2
546..
.3...
.2...
.1...
...

correct output
-
-

user output
-
-