CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:siar
Submission time:2020-10-04 15:08:58 +0300
Language:Python3 (CPython3)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
Test results
testverdicttimegroup
#1ACCEPTED0.03 s1, 2details
#2ACCEPTED0.03 s1, 2details
#3ACCEPTED0.03 s1, 2details
#4ACCEPTED0.03 s1, 2details
#5ACCEPTED0.03 s1, 2details
#6ACCEPTED0.03 s1, 2details
#7ACCEPTED0.03 s1, 2details
#8ACCEPTED0.03 s1, 2details
#9ACCEPTED0.03 s1, 2details
#10ACCEPTED0.03 s1, 2details
#11ACCEPTED0.03 s2details
#12ACCEPTED0.03 s2details
#13ACCEPTED0.03 s2details
#14ACCEPTED0.03 s2details
#15ACCEPTED0.03 s2details
#16ACCEPTED0.03 s2details
#17ACCEPTED0.03 s2details
#18ACCEPTED0.03 s2details
#19ACCEPTED0.03 s2details
#20ACCEPTED0.03 s2details
#210.03 s2details
#22ACCEPTED0.03 s2details
#230.03 s2details
#24ACCEPTED0.03 s2details
#25ACCEPTED0.03 s2details
#26ACCEPTED0.03 s2details
#27ACCEPTED0.03 s2details
#28ACCEPTED0.03 s2details
#290.03 s2details
#300.03 s2details
#310.03 s1, 2details

Code

import copy
import sys



no_of_lines = int(input())
dice_patterns = ""
dice_pattern = []

no_of_liness = no_of_lines*6


for i in range(no_of_liness):
    dice_patterns += input()+"\n"



lines = [x for x in dice_patterns.split()]
pattern_strings = []
for i in range(0, len(lines), 5):
    dice_pattern.append("\n".join(lines[i:i+5]))

#print(dice_pattern)


#Strip the string then append it to the list



#dice_pattern.append(dice_patterns.strip(" ").split(" "))

#
# for idx, item in enumerate(dice_pattern):
#     if item == 4:
#         dice_pattern[idx] = 44


#print(dice_patterns)
#print(dice_pattern)
#print(dice_pattern)

#dice_pattern = data = sys.stdin.read()



def rotate_dice(dice):
    rotated = []
    for i in range(len(dice[0])):
        rotated.append([row[i] for row in dice][::-1])
    return rotated

def parse_dice(dice_string):
    dice = [[0, 0, 0, 0, 0] for _ in range(5)]
    lines = [line.strip() for line in dice_string.split("\n") if line and not line.startswith(".....")]
    for i, line in enumerate(lines):
        line = line.replace(".", "0")
        for j, c in enumerate(line):
            dice[i][j] = int(c)

    return dice


def get_opposite_sides(dice):
    dice_copy = copy.deepcopy(dice)
    side_tuples = []
    for i, row in enumerate(dice):
        nonzeros = [(col_idx, nz) for col_idx, nz in enumerate(row) if nz > 0]
        if len(nonzeros) == 4:          # if we find 4 in a row, save 1, 3 and 2, 4 as oppsite sides
            side_tuples.append([nonzeros[0][1], nonzeros[2][1]])
            side_tuples.append([nonzeros[1][1], nonzeros[3][1]])
            dice_copy[i] = [0, 0, 0, 0, 0]
        elif len(nonzeros) == 3:
            side_tuples.append([nonzeros[0][1], nonzeros[2][1]])
            dice_copy[i][nonzeros[0][0]] = 0
            dice_copy[i][nonzeros[2][0]] = 0

    for col_idx in range(len(dice[0])):
        col = [row[col_idx] for row in dice]
        nonzeros = [(row_idx, nz) for row_idx, nz in enumerate(col) if nz > 0]
        if len(nonzeros) == 4:
            side_tuples.append([nonzeros[0][1], nonzeros[2][1]])
            side_tuples.append([nonzeros[1][1], nonzeros[3][1]])
            for row in dice_copy:
                row[col_idx] = 0
        elif len(nonzeros) == 3:
            side_tuples.append([nonzeros[0][1], nonzeros[2][1]])
            dice_copy[nonzeros[0][0]][col_idx] = 0
            dice_copy[nonzeros[2][0]][col_idx] = 0

    if len(side_tuples) == 3:
        return side_tuples

    for i, row in enumerate(dice_copy):
        for j, col in enumerate(row):
            if col == 0:
                continue
            dice_copy[i][j] = 0
            shape = (len(dice), len(dice[0]))
            if i+1 < shape[0] and j+2 < shape[1] and dice_copy[i+1][j+2] != 0:      # look one down and two to the right for another number
                side_tuples.append([col, dice[i+1][j+2]])
                dice_copy[i+1][j+2] = 0
            elif i+2 < shape[0] and j+1 < shape[1] and dice_copy[i+2][j+1] != 0:    # look two down and one to the right for another number
                side_tuples.append([col, dice[i+2][j+1]])
                dice_copy[i+2][j+1] = 0
            elif i + 2 < shape[0] and j + 2 < shape[1] and dice_copy[i + 2][j + 2] != 0:  # look two down and one to the right for another number
                side_tuples.append([col, dice[i + 2][j + 2]])
                dice_copy[i + 2][j + 2] = 0
            elif i + 3 < shape[0] and j + 2 < shape[1] and dice_copy[i + 3][j + 2] != 0:  # look two down and one to the right for another number
                side_tuples.append([col, dice[i + 3][j + 2]])
                dice_copy[i + 3][j + 2] = 0
    if len(side_tuples) < 3:
        side_tuples.append(list({1, 2, 3, 4, 5, 6}.difference(sum(side_tuples, []))))
    if len(side_tuples) == 3:
        return side_tuples
    else:
        return get_opposite_sides(rotate_dice(dice))

# dice_pattern = [
#     '''
#         25...
#         .643.
#         .1...
#         .....
#         .....
# '''
#     ,
# '''
#         .3...
#         6412.
#         5....
#         .....
#         .....
# '''
#    ,
# '''
#         ..2..
#         1653.
#         4....
#         .....
#         .....
#
# '''
#   ,
# '''
#         .....
#         ...5.
#         .3264
#         ...1.
#         .....
# '''
# ]


allList = []

for i in range(len(dice_pattern)):
    d = parse_dice(dice_pattern[i])
    opposite_sides = get_opposite_sides(d)
    opposite_sides[0].sort()
    opposite_sides[1].sort()
    opposite_sides[2].sort()
    opposite_sides.sort()
    # print(dice_pattern[i])
    allList.append(opposite_sides)
    #print(opposite_sides)


#print(allList)

#indexx = allList.index([[1, 5], [2, 4], [3, 6]])
#print(indexx)

indices = []

def get_index_positions(list_of_elems, element):

    index_pos_list = []
    index_pos = 0
    while True:
        try:
            # Search for item in list from indexPos to the end of list
            index_pos = list_of_elems.index(element, index_pos)
            # Add the index position in list
            index_pos_list.append(index_pos+1)
            index_pos += 1
        except ValueError as e:
            break
    return index_pos_list




for i, everyElement in enumerate(allList):

    indexxes = get_index_positions(allList, everyElement)
    indexxes.remove(i+1)
    indices.append(indexxes)

#print(indices)


for everyElements in indices:

    if not everyElements:
        print("-")
    else:
        print(*everyElements)


#for everyElement in allList:

    #if allList.index(everyElement) == True:







#
# for i, listt in enumerate(allList):
#     ddar = []
#     for n in range(len(allList)):
#
#         if allList[n] == listt and n != i:
#             all
#             print(" same with", n+1)
#
#         elif n != i and allList[n] != listt:
#             print("s")
#
#





# for everyElement in allList:
#
#
#
#
#
#
#
#     for i in range(len(allList)):
#
#         if allList[i] == everyElement and allList.index(everyElement) != i:
#             print(everyElement, " same with", i)

        #elif allList.index(everyElement) != i:
            #continue
        #else:
            #print(".")

    # for n in range(len(dice_pattern)):
    #     textList2 = []
    #
    #     try:
    #         dd = parse_dice(dice_pattern[n])
    #         opposite_sidess = get_opposite_sides(dd)
    #     #print(dice_pattern[i])
    #     #print(opposite_sides)
    #         opposite_sidess[0].sort()
    #         opposite_sidess[1].sort()
    #         opposite_sidess[2].sort()
    #         opposite_sidess.sort()
    #
    #         #print(opposite_sidess)
    #         #print(opposite_sides)
    #         #print("22222")
    #
    #         if opposite_sides == opposite_sidess and i != n:
    #             a = n+1
    #             textList2.append(a)
    #
    #



        #except IndexError:
            #continue

        #textList.append([textList2])
        # elif i == n:
        # textList2.append("-")

    #textList2.append(textList)

#print(textList)
#print(textList2)

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

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

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

user output
9
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
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: 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
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: 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:

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:

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

Test 31

Group: 1, 2

Verdict:

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

correct output
-
-

user output
2
1