CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:xnor
Submission time:2020-09-29 15:13:01 +0300
Language:Rust
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED35
#2ACCEPTED65
Test results
testverdicttimegroup
#1ACCEPTED0.01 s1, 2details
#2ACCEPTED0.01 s1, 2details
#3ACCEPTED0.01 s1, 2details
#4ACCEPTED0.01 s1, 2details
#5ACCEPTED0.01 s1, 2details
#6ACCEPTED0.01 s1, 2details
#7ACCEPTED0.01 s1, 2details
#8ACCEPTED0.01 s1, 2details
#9ACCEPTED0.01 s1, 2details
#10ACCEPTED0.01 s1, 2details
#11ACCEPTED0.01 s2details
#12ACCEPTED0.01 s2details
#13ACCEPTED0.01 s2details
#14ACCEPTED0.01 s2details
#15ACCEPTED0.01 s2details
#16ACCEPTED0.01 s2details
#17ACCEPTED0.01 s2details
#18ACCEPTED0.01 s2details
#19ACCEPTED0.01 s2details
#20ACCEPTED0.01 s2details
#21ACCEPTED0.01 s2details
#22ACCEPTED0.01 s2details
#23ACCEPTED0.01 s2details
#24ACCEPTED0.01 s2details
#25ACCEPTED0.01 s2details
#26ACCEPTED0.01 s2details
#27ACCEPTED0.01 s2details
#28ACCEPTED0.01 s2details
#29ACCEPTED0.01 s2details
#30ACCEPTED0.01 s2details
#31ACCEPTED0.01 s1, 2details

Code

use std::io::{self, BufRead};

fn main() {
    let mut lines = io::BufReader::new(io::stdin()).lines();

    let n: i32 = lines.next().unwrap().unwrap().parse().unwrap();

    let mut dices = vec![
        Dice {
            u: 1,
            d: 0,
            f: 0,
            b: 0,
            l: 0,
            r: 0,
        };
        n as usize
    ];

    for i in 0..n as usize {
        let mut grid = [[0; 5]; 5];
        let (mut cx, mut cy) = (0, 0);
        for y in 0..5 {
            let line = lines.next().unwrap().unwrap();
            for (x, c) in line.bytes().enumerate() {
                if c == b'1' {
                    cx = x;
                    cy = y;
                } else if c != b'.' {
                    grid[y][x] = (c - b'0') as i32;
                }
            }
        }
        if i != n as usize - 1 {
            lines.next().unwrap().unwrap();
        }
        build_dice(&mut grid, cx, cy, &mut dices[i]);
        let m = dices[i].f.min(dices[i].l).min(dices[i].b).min(dices[i].r);
        while dices[i].f != m {
            let tmp = dices[i].f;
            dices[i].f = dices[i].r;
            dices[i].r = dices[i].b;
            dices[i].b = dices[i].l;
            dices[i].l = tmp;
        }
    }
    
    for i in 0..n as usize {
        let mut b = false;

        for j in 0..n as usize {
            if j != i && dices[i] == dices[j] {
                print!("{} ", j + 1);
                b = true;
            }
        }

        if !b {
            println!("-");
        } else {
            println!();
        }
    }
}

fn build_dice(grid: &mut [[i32; 5]; 5], cx: usize, cy: usize, dice: &mut Dice) {
    if cy < 4 && grid[cy + 1][cx] != 0 {
        dice.f = dice.d;
        dice.d = dice.b;
        dice.b = dice.u;
        dice.u = grid[cy + 1][cx];

        grid[cy + 1][cx] = 0;
        build_dice(grid, cx, cy + 1, dice);

        let tmp = dice.u;
        dice.u = dice.b;
        dice.b = dice.d;
        dice.d = dice.f;
        dice.f = tmp;
    }

    if cy > 0 && grid[cy - 1][cx] != 0 {
        dice.b = dice.u;
        std::mem::swap(&mut dice.d, &mut dice.f);
        std::mem::swap(&mut dice.r, &mut dice.l);
        dice.u = grid[cy - 1][cx];

        grid[cy - 1][cx] = 0;

        rotate_grid_right(grid);
        rotate_grid_right(grid);

        build_dice(grid, 4 - cx, 5 - cy, dice);

        rotate_grid_right(grid);
        rotate_grid_right(grid);

        std::mem::swap(&mut dice.d, &mut dice.f);
        std::mem::swap(&mut dice.b, &mut dice.u);
        std::mem::swap(&mut dice.r, &mut dice.l);
    }

    if cx > 0 && grid[cy][cx - 1] != 0 {
        dice.l = dice.b;
        dice.b = dice.u;
        let tmp = dice.r;
        dice.r = dice.f;
        dice.f = dice.d;
        dice.d = tmp;
        dice.u = grid[cy][cx - 1];

        grid[cy][cx - 1] = 0;
        rotate_grid_right(grid);

        build_dice(grid, cy, 5 - cx, dice);

        rotate_grid_right(grid);
        rotate_grid_right(grid);
        rotate_grid_right(grid);

        let tmp = dice.l;
        dice.l = dice.u;
        dice.u = dice.b;
        dice.b = tmp;
        let tmp = dice.d;
        dice.d = dice.f;
        dice.f = dice.r;
        dice.r = tmp;
    }

    if cx < 4 && grid[cy][cx + 1] != 0 {
        let tmp = dice.f;
        dice.f = dice.d;
        dice.d = dice.l;
        dice.l = tmp;
        dice.r = dice.b;
        dice.b = dice.u;
        dice.u = grid[cy][cx + 1];

        grid[cy][cx + 1] = 0;

        rotate_grid_right(grid);
        rotate_grid_right(grid);
        rotate_grid_right(grid);

        build_dice(grid, 4 - cy, cx + 1, dice);

        rotate_grid_right(grid);

        let tmp = dice.r;
        dice.r = dice.u;
        dice.u = dice.b;
        dice.b = tmp;
        let tmp = dice.l;
        dice.l = dice.d;
        dice.d = dice.f;
        dice.f = tmp;
    }
}

fn rotate_grid_right(grid: &mut [[i32; 5]; 5]) {
    let mut new_grid = [[0; 5]; 5];

    for y in 0..5 {
        for x in 0..5 {
            new_grid[y][x] = grid[x][4 - y];
        }
    }

    *grid = new_grid;
}

#[derive(PartialEq, Eq, Clone)]
struct Dice {
    u: i32,
    d: i32,
    l: i32,
    r: i32,
    f: i32,
    b: i32,
}

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3
-
1

user output

-

Test 2

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 3

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3 5
4
1 5
2
1 3

user output
3 5 

1 5 

1 3 

Test 4

Group: 1, 2

Verdict: ACCEPTED

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

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 5

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 6

Group: 1, 2

Verdict: ACCEPTED

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

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 7

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3
4 5
1
2 5
2 4

user output

4 5 

2 5 
2 4 

Test 8

Group: 1, 2

Verdict: ACCEPTED

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

correct output
4 5
3
2
1 5
1 4

user output
4 5 


1 5 
1 4 

Test 9

Group: 1, 2

Verdict: ACCEPTED

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

correct output
2 3
1 3
1 2
5
4

user output
2 3 
1 3 
1 2 


Test 10

Group: 1, 2

Verdict: ACCEPTED

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

correct output
5
3 4
2 4
2 3
1

user output

3 4 
2 4 
2 3 

Test 11

Group: 2

Verdict: ACCEPTED

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

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

user output

4 6 8 
7 10 
2 6 8 
-
...

Test 12

Group: 2

Verdict: ACCEPTED

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

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

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

Test 13

Group: 2

Verdict: ACCEPTED

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

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

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

Test 14

Group: 2

Verdict: ACCEPTED

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

Test 15

Group: 2

Verdict: ACCEPTED

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

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

user output
5 7 9 10 

4 6 
3 6 
1 7 9 10 
...

Test 16

Group: 2

Verdict: ACCEPTED

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

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

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

Test 17

Group: 2

Verdict: ACCEPTED

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

Test 18

Group: 2

Verdict: ACCEPTED

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

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

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

Test 19

Group: 2

Verdict: ACCEPTED

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
2 5 6 7 8 9 
1 5 6 7 8 9 
4 10 
3 10 
1 2 6 7 8 9 
...
Truncated

Test 20

Group: 2

Verdict: ACCEPTED

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

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

user output
4 7 8 
6 9 10 

1 7 8 

...

Test 21

Group: 2

Verdict: ACCEPTED

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

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

user output
3 7 11 16 

1 7 11 16 
5 19 
4 19 
...
Truncated

Test 22

Group: 2

Verdict: ACCEPTED

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
7 10 11 17 20 
12 
4 9 13 15 18 
3 9 13 15 18 
8 14 16 
...
Truncated

Test 23

Group: 2

Verdict: ACCEPTED

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
5 12 13 15 18 
16 20 
6 8 14 
9 19 
1 12 13 15 18 
...
Truncated

Test 24

Group: 2

Verdict: ACCEPTED

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
2 6 11 12 13 
1 6 11 12 13 
5 16 18 
7 14 
3 16 18 
...
Truncated

Test 25

Group: 2

Verdict: ACCEPTED

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

Test 26

Group: 2

Verdict: ACCEPTED

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
8 10 20 
3 17 18 19 
2 17 18 19 
14 15 
6 7 9 13 
...
Truncated

Test 27

Group: 2

Verdict: ACCEPTED

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

Test 28

Group: 2

Verdict: ACCEPTED

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

Test 29

Group: 2

Verdict: ACCEPTED

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
5 8 9 14 17 
3 10 16 
2 10 16 
13 15 19 
1 8 9 14 17 
...
Truncated

Test 30

Group: 2

Verdict: ACCEPTED

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

Test 31

Group: 1, 2

Verdict: ACCEPTED

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

correct output
-
-

user output
-
-