Arpakuutiot

import re
from itertools import chain

# all possible "shapes" of the dice when flattened out,
# i.e. the indexes of numbers in a 3x4 string
# as pairs of values on opposite sides from each other
number_placements = [
[(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)]
]
# dictionary mapping from set of all numbers in the shape to list of pairs of values
shapes = {frozenset(chain.from_iterable(v)): v for v in number_placements}

# flips a 5x5 input string
# this ensures that mirror images of the shapes will be matched
def flip(string):
    # converts string into list of reversed rows
    aslist = [string[i:i+5][::-1] for i in range(0, 25, 5)]
    
    return "".join(aslist)

# rotates 5x5 string by 90 degrees
def rotate(string):
    # converts string into list of rows
    aslist = [string[i:i+5] for i in range(0, 25, 5)]
    result = [""]*5
    # iterates over the reversed list and adds characters of every row to the result
    for row in aslist[::-1]:        
        result = [result[i] + row[i] for i in range(5)]
    
    return "".join(result)

# formats the (5x5) string into a 3x4 block
def parse(string):
    # any numbers high enough will do
    # just ensures that the min function will change the values
    x, y = 6, 6
    for i, c in enumerate(string):
        # x and y are the first row and column of the string that have any numbers
        if c != ".": x, y = min(i%5, x), min(i//5, y)
    # removes the unnecessary rows and columns
    # if the dice is in "4*3" format, this will just trip a KeyError later on
    string = string[5*y:5*(y+4)]
    aslist = [string[i:i+5] for i in range(0, len(string), 5)]
    result = [row[x:x+3] for row in aslist]
    
    return "".join(result)

def find_shape(string):
    # loops through possible rotations until match is found
    for r in range(4):
        # considers both the non-mirrored and mirrored versions
        for copy in (string, flip(string)):
            # EAFP
            try:
                s = parse(copy)
                # key for the shapes dictionary
                indices = frozenset(s.index(str(i)) for i in range(1,7))
                pairs = to_pairs(shapes[indices], s)

            # wrong rotation/mirror of the shape wont be found
            except KeyError: pass
            # converting a dot to int
            except ValueError: pass
            else:
                return pairs

        string = rotate(string)
    # something went wrong
    raise ValueError

def to_pairs(pairshapes, s):
    result = set()
    for f1, f2 in pairshapes:
        # adds the value set to the set
        result.add(frozenset((int(s[f1]), int(s[f2]))))
    return result


n = int(input())

dice = []
for i in range(n):
    
    die = "".join([input() for _ in range(6)])
    dice.append(find_shape(die))

for i, die in enumerate(dice):
    same = []
    for j, other in enumerate(dice):
        # same values, but they arent the same dice
        if (i != j) and (die == other):
            same.append(str(j + 1))
    if same: print(" ". join(same))
    else: print("-")

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 1, 2

Verdict: ACCEPTED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 1, 2

Verdict: ACCEPTED

Test 7

Group: 1, 2

Verdict: ACCEPTED

Test 8

Group: 1, 2

Verdict: ACCEPTED

Test 9

Group: 1, 2

Verdict: ACCEPTED

Test 10

Group: 1, 2

Verdict: ACCEPTED

Test 11

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 12

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 13

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 14

Group: 2

Verdict: ACCEPTED

Test 15

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 16

Group: 2

Verdict: ACCEPTED

Test 17

Group: 2

Verdict: ACCEPTED

Test 18

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 19

Group: 2

Verdict: ACCEPTED

Test 20

Group: 2

Verdict: ACCEPTED

Test 21

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 22

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 23

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 24

Group: 2

Verdict: ACCEPTED

Test 25

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 26

Group: 2

Verdict: ACCEPTED

Test 27

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 28

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 29

Group: 2

Verdict: WRONG ANSWER

Test 30

Group: 2

Verdict: RUNTIME ERROR

Traceback (most recent call last):
  File "input/code.py", line 94, in <module>
    dice.a...

Test 31

Group: 1, 2

Verdict: WRONG ANSWER