CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:jmarttinen
Submission time:2020-09-30 12:16:15 +0300
Language:Python3 (PyPy3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#1ACCEPTED0.07 s1, 2details
#2ACCEPTED0.07 s1, 2details
#3ACCEPTED0.07 s1, 2details
#4ACCEPTED0.07 s1, 2details
#5ACCEPTED0.07 s1, 2details
#6ACCEPTED0.07 s1, 2details
#7ACCEPTED0.07 s1, 2details
#8ACCEPTED0.07 s1, 2details
#9ACCEPTED0.07 s1, 2details
#10ACCEPTED0.07 s1, 2details
#110.07 s2details
#120.08 s2details
#130.07 s2details
#14ACCEPTED0.07 s2details
#15ACCEPTED0.07 s2details
#16ACCEPTED0.07 s2details
#17ACCEPTED0.07 s2details
#180.07 s2details
#19ACCEPTED0.07 s2details
#20ACCEPTED0.07 s2details
#210.08 s2details
#220.08 s2details
#230.08 s2details
#24ACCEPTED0.08 s2details
#250.08 s2details
#26ACCEPTED0.08 s2details
#270.08 s2details
#280.08 s2details
#290.08 s2details
#300.08 s2details
#310.07 s1, 2details

Code

# xxx   .x.     .xx     .xx     xx       x       x       x      x        x
# .x.   xxx     xx.     .x.     x      xxx      xx      xx      xx      xx
# .x.   .x.     .x.     xx.     x       x      xx       x      xx      xx 
# .x.   .x.     .x.     .x.    xx       x       x      xx       x      x 

import re
from itertools import chain
# in the same order as the above ones
# from elements to ordered element pairs
shapes = {
(0,1,2,4,7,10): [(0,2),(1,7),(4,10)],
(1,3,4,5,7,10): [(1,7),(3,5),(4,10)],
(1,2,3,4,7,10): [(1,7),(2,3),(4,10)],
(1,2,4,6,7,10): [(1,7),(2,6),(4,10)],
(1,2,4,7,9,10): [(1,7),(2,9),(4,10)],
(2,3,4,5,7,10): [(2,7),(3,5),(4,10)],
(2,4,5,6,7,10): [(2,7),(5,6),(4,10)],
(2,4,5,7,9,10): [(2,7),(5,9),(4,10)],
(1,4,5,6,7,10): [(1,7),(5,6),(4,10)],
(2,4,5,6,7,9): [(2,7),(5,6),(4,9)]
}
shapes2 = {frozenset(k): v for k, v in shapes.items()}
shapes = [
[(0,2),(1,7),(4,10)],
[(1,7),(3,5),(4,10)],
[(1,7),(2,3),(4,10)],
[(1,7),(2,6),(4,10)],
[(1,7),(2,9),(4,10)],
[(2,7),(3,5),(4,10)],
[(2,7),(5,6),(4,10)],
[(2,7),(5,9),(4,10)],
[(1,7),(5,6),(4,10)],
[(2,7),(5,6),(4,9)]
]
shapes2 = {frozenset(chain.from_iterable((v))): v for v in shapes}

def mirror(string):
    size = int(len(string)**0.5)
    aslist = [string[i:i+size][::-1] for i in range(0, len(string), size)]
    
    """result = ""

    for row in aslist:
        result += row[::-1]
    return result"""
    return "".join(aslist)

def rotate(string):
    """Rotates a 'square shaped' string by 90 degrees
    for example abcd -> cadb
    """
    # transforms string into list of 'rows'
    # added this first line for reusability
    size = int(len(string)**0.5)
    aslist = [string[i:i+size] for i in range(0, len(string), size)]
    result = [""]*size
    # loops through the reversed list and adds characters one by one to the result rows
    for row in aslist[::-1]:
        result = [result[i] + row[i] for i in range(size)]
    return "".join(result)

def parse(string):
    """Parses the input string somewhat by locking the """
    # any numbers higher than the max possible
    # for a 5x5 square would do
    x, y = 6,6
    for i, c in enumerate(string):
        if c != ".":
            # x and y should be the first sides of the square, that cannot be removed without destroying the formatting
            # e.g. in ...X., the x coordinate is 3 (indexing starts at 0) 
            x, y = min(x, i%5), min(y, i//5)
    # removes unnecessary lines
    # because a die is always 4*3, and it doesn't matter if this happens
    # to be rotated wrong because it will just cause a ValueError in the try clause
    string = string[5*y:5*(y+4)]
    aslist = [string[i:i+5] for i in range(0, len(string), 5)]
    result = ""
    for row in aslist:
        result += row[x:x+3]
    
    return result


def die_shape(string):
    # loops through possible rotations until match is found
    for r in range(4):
        # considers both non-mirrored and mirrored version
        for copy in (string, mirror(string)):
            try:
                s = parse(copy)
                # raises ValueError if any of the indices are dots
                indices = frozenset(s.index(str(i)) for i in range(1, 7))
                # compares the set to the dictionary
                pairs = to_pairs(shapes2[indices], s)
            except KeyError:
                pass
            except ValueError:
                pass
            else:
                return pairs
               
        string = rotate(string)

def to_pairs(pairs, s):
    result = set()
    # loops through the pairs and adds a frozenset containing the 
    # opposite faces of the dice
    # two dice are the same iff they have all the same opposites
    for n1, n2 in pairs:
        result.add(frozenset((int(s[n1]), int(s[n2]))))
    return result


n = int(input())
inputs = []
try:
    for _ in range(6*n):
        inp = input()
        inputs.append(inp)
        # captures the empty line
        # input()

    dice = ["".join(inputs[i:i+5]) for i in range(0, 6*n, 6)]
except EOFError:
        # cses seems to be causing an eof error for some reason
        dice = [inputs[i:i+25] for i in range(0, 6*n, 25)]


if dice:
    dice = list(map(die_shape, dice))


    for i, die in enumerate(dice):
        #print(die)
        same = []
        for j, other in enumerate(dice):
            if (i != j) and (die == other):
                same.append(str(j+1))

        if same:
            print(" ".join(same))
        else:
            print("-")

    # string = "".join([input() for _ in range(5)])

    # for r in range(4):
    #     string = rotate(string)
    #     print(string)

Test details

Test 1

Group: 1, 2

Verdict: ACCEPTED

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

correct output
3
-
1

user output
3
-
1

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
3
4 5
1
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
4
1 5
2
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
3
2
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
3
4 5
1
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
3
2
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
3
4 5
1
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
3
2
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
5
4

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
5
3 4
2 4
2 3
1

Test 11

Group: 2

Verdict:

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

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

user output
9
4 6
7 10
2 6
-
...

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

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
4 10
5 6 7 8
-
1 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
8
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
...

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
3 8 9
5 6
1 8 9
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
5
1 7 8
3
...

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 3 6 7 11 16
1 3 6 7 11 16
1 2 6 7 11 16
5 19
4 19
...
Truncated

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

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

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 17 19
1 3 5 7 17 19
1 2 5 7 17 19
8 10 11 14
1 2 3 7 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:

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
19
1 2 3 6 9 20
...
Truncated

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

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

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

Test 31

Group: 1, 2

Verdict:

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

correct output
-
-

user output
2
1