CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:jmarttinen
Submission time:2020-09-30 12:07:27 +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 ValueError:
dice = []
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