Arpakuutiot

from itertools import chain

class Die:

    def __init__(self, pairs):
        self.pairs = set()
        self.parse(pairs)
        

    def __eq__(self, other):
        return self.pairs == other.pairs

    def parse(self, text):
        indices = {k: text.index(str(k)) for k in range(1, 7)}
        self.cantbe = {k: {k} for k in range(1,7)}
        while len(self.pairs) < 3:
            for k, v in indices.items():
                for k2,v2 in indices.items():
                    if (abs(v - v2) == 2) & ((v // 5) == (v2 // 5)):
                        self.addpair(k,k2)
                    elif (abs(v - v2) == 10):
                        self.addpair(k,k2)
                    elif (abs(v%5 - v2%5) <= 1) and (abs(v - v2) <= 1):
                        self.cantbe[k].add(k2)
                        self.cantbe[k2].add(k)
                if len(self.pairs) == 2:
                    rest = set(range(1,7)) - set(chain(*self.pairs))
                    self.addpair(*rest)
                elif len(self.cantbe[k]) == 5:
                    last = set(range(1,7)) - set(chain(*self.cantbe[k]))
                    self.addpair(k, last)

    def addpair(self, p1, p2):
        self.pairs.add(frozenset((p1, p2)))
        for k in self.cantbe.keys():
            self.cantbe[k].add(p1)
            self.cantbe[k].add(p2)


n = int(input())
dice = []
for _ in range(n):
    uinput = "".join([input() for row in range(6)])
    die = Die(uinput)
    dice.append(die)
    
for die in dice:
    same = []
    for i, other in enumerate(dice):
        if die == other and die is not other:
            same.append(str(i + 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: TIME LIMIT EXCEEDED

Test 12

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 13

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 14

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 15

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 16

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 17

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 18

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 19

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 20

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 21

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 22

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 23

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 24

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 25

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 26

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 27

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 28

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 29

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 30

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 31

Group: 1, 2

Verdict: WRONG ANSWER