CSES - Datatähti 2021 alku - Results
Submission details
Task:Arpakuutiot
Sender:Ugi
Submission time:2020-10-10 19:04:32 +0300
Language:Python3 (PyPy3)
Status:READY
Result:100
Feedback
groupverdictscore
#1ACCEPTED35
#2ACCEPTED65
Test results
testverdicttimegroup
#1ACCEPTED0.05 s1, 2details
#2ACCEPTED0.05 s1, 2details
#3ACCEPTED0.05 s1, 2details
#4ACCEPTED0.05 s1, 2details
#5ACCEPTED0.05 s1, 2details
#6ACCEPTED0.05 s1, 2details
#7ACCEPTED0.05 s1, 2details
#8ACCEPTED0.05 s1, 2details
#9ACCEPTED0.06 s1, 2details
#10ACCEPTED0.05 s1, 2details
#11ACCEPTED0.06 s2details
#12ACCEPTED0.06 s2details
#13ACCEPTED0.06 s2details
#14ACCEPTED0.06 s2details
#15ACCEPTED0.06 s2details
#16ACCEPTED0.06 s2details
#17ACCEPTED0.06 s2details
#18ACCEPTED0.06 s2details
#19ACCEPTED0.06 s2details
#20ACCEPTED0.06 s2details
#21ACCEPTED0.07 s2details
#22ACCEPTED0.06 s2details
#23ACCEPTED0.07 s2details
#24ACCEPTED0.07 s2details
#25ACCEPTED0.06 s2details
#26ACCEPTED0.06 s2details
#27ACCEPTED0.07 s2details
#28ACCEPTED0.06 s2details
#29ACCEPTED0.06 s2details
#30ACCEPTED0.07 s2details
#31ACCEPTED0.05 s1, 2details

Code

def pret(solution):
    return [int("".join([x.replace(".","") for x in solution if x != "....."]))]
def fit(dice):
    for x in dice:
        if len([z for z in x if z != "."]) == 3:
            li1 = "".join([z for z in x if z != "."])
        elif len([z for z in x if z != "."]) == 4:
            li2 = "".join([z for z in x if z != "."][::-1])
    for y in [list(map(lambda l: l[t], dice)) for t in range(5)]:
        if len([z for z in y if z != "."]) == 3:
            li1 = "".join([z for z in y if z != "."])
        if len([z for z in y if z != "."]) == 4:
            li2 = "".join([z for z in y if z != "."])
    ind = li2.index(li1[1])
    li3 = [li2[(ind+1)%4],li2[(ind+2)%4],li2[(ind+3)%4]]
    return [li1+"..","."+li3[0]+"...","."+li3[1]+"...","."+li3[2]+"...","....."]
def fix(dice,rc,i):
    if rc == "r":
        for a,e in enumerate(dice[i-1]):
            if e != ".":
                ind = a
                break
        u = dice[i+1].replace(".","")
        dice[i+1] = "....."
        dice[i+1] = dice[i+1][:ind]+u+dice[i+1][ind:]
        return dice
    if rc == "c":
        for a,e in enumerate(list(map(lambda l: l[i-1], dice))):
            if e != ".":
                ind = a
                u = e
                break
        dice[ind] = dice[ind][:i-1]+"."+dice[ind][i:]
        for z,y in enumerate(list(map(lambda l: l[i+1], dice))):
            if y != ".":
                ind2 = z
                break
        dice[ind2] = dice[ind2][:i-1]+u+dice[ind2][i:]
        return dice
def force(dice):
    if len([len("".join(z).replace(".","")) for z in [dice[x] for x in range(5)] if len("".join(z).replace(".","")) == 3]) == 2:
        return force(twothree(dice))
    if len([len("".join(z).replace(".","")) for z in [list(map(lambda l: l[t], dice)) for t in range(5)] if len("".join(z).replace(".","")) == 3]) == 2:
        return force(twothree(dice))
    for i,e in enumerate([dice[y] for y in range(5)]):
        if len([x for x in e if x != "."]) == 4:
            return fix(dice,"r",i)
    for i,e in enumerate([list(map(lambda l: l[t], dice)) for t in range(5)]):
        if len([x for x in e if x != "."]) == 4:
            return fix(dice,"c",i)
    for i,row in enumerate(dice):
        if len([z for z in row if z != "."]) == 3:
            forth = (dice[(i-2)%5]+dice[(i+2)%5]).replace(".","")
            if len(forth) == 0:
                for q,spot in enumerate(row):
                    if spot == ".":
                        if len([z for z in list(map(lambda l: l[q], dice)) if z != "."]) == 1:
                            forth = [z for z in list(map(lambda l: l[q], dice)) if z != "."][0]
                            dice = [x.replace(forth,".") for x in dice]
                            break
            else:
                dice[i-2],dice[(i+2)%5] = ".....","....."
            if row[:2] == "..":
                dice[i] = "."+forth+row[2:]
            elif row[0] == ".":
                dice[i] = forth+row[1:]
            elif row[-2:] == "..":
                dice[i] = row[:-2]+forth+"."
            dice = fix(dice,"r",i)
            return dice
    for i,column in enumerate([list(map(lambda l: l[t], dice)) for t in range(5)]):
        if len([z for z in column if z != "."]) == 3:
            return force(rotate(dice[::-1]))
    return force(only2(dice))
def twothree(dice):
    for i,row in enumerate([dice[t] for t in range(5)]):
        if len([x for x in row if x != "."]) == 3:
            for z,spot in enumerate(row):
                if spot == ".":
                   continue
                if i > 2:
                    return twothree([dice[1],dice[2],dice[3],dice[4],"....."])
                if i < 2:
                    return twothree([".....",dice[0],dice[1],dice[2],dice[3]])
                if dice[(i+1)%5][z] == "." and dice[i-1][z] == ".":
                    forth = spot
                    dice[i] = dice[i][:z]+"."+dice[i][z+1:]
                    if dice[i+1][z+2] != ".":
                        dice[i+2] = (z+2)*"."+forth+(2-z)*"."
                    elif dice[i-1][z+2] != ".":
                        dice[i-2] = (z+2)*"."+forth+(2-z)*"."
                    return dice
                else:
                    forth = row[z+2]
                    dice[i] = dice[i].replace(forth,".")
                    if dice[i+1][z-2] != ".":
                        dice[i+2] = (z-2)*"."+forth+(6-z)*"."
                    elif dice[i-1][z-2] != ".":
                        dice[i-2] = (z-2)*"."+forth+(6-z)*"."
                    return dice
    for i,column in enumerate([list(map(lambda l: l[t], dice)) for t in range(5)]):
        return twothree(list("".join(map(lambda l: l[t], dice))[::-1] for t in range(5)))
def rotate(dice):
    return list("".join(map(lambda l: l[t], dice)) for t in range(5))
def only2(dice):
    f=lambda a: (abs(a)+a)/2
    for i,row in enumerate(dice):
        if len([z for z in row if z != "."]) != 1:
            continue
        for z,spot in enumerate(row):
            if spot == ".":
                continue
            dice[i] == "....."
            if dice[i+1][(z+1)%4] != "." or dice[i-1][(z+1)%4] != ".":
                dice[i] = (z+1)*"."+spot+(3-z)*"."
                return dice
            if dice[i+1][z-1] != "." or dice[i-1][z-1] != ".":
                dice[i] = (z-1)*"."+spot+(5-z)*"."
                return dice
    for i,column in enumerate([list(map(lambda l: l[t], dice)) for t in range(5)]):
        if len([z for z in column if z != "."]) != 1:
            continue
        for z,spot in enumerate(column):
            if spot == ".":
                continue
            f=lambda a: (abs(a)+a)/2
            dice[z] = dice[z][:i]+"."+dice[z][i+1:]
            if dice[z-1][(i+1)%4] != "." or dice[z-1][i-1] != ".":
                dice[z-1] = dice[z-1][:int(f(i))]+spot+dice[z-1][i+1:]
                return dice
            if dice[z+1][(i+1)%4] != "." or dice[z+1][i-1] != ".":
                dice[z+1] = dice[z+1][:int(f(i))]+spot+dice[z+1][i+1:]
                return dice
def turnover(dice):
    return list(turn(turn(dice)))
def turn3(dice):
    lis = []
    pi = list(shift(dice))
    lis += pret(pi)
    pi = list(shift(pi))
    lis += pret(pi)
    pi = list(shift(pi))
    lis += pret(pi)
    return lis,pi
def check(dice):
    lit = []
    lit += pret(dice)
    lis,pi = turn3(dice)
    lit += lis
    pi = turnover(pi)
    lit += pret(pi)
    lis,pi = turn3(pi)
    lit += lis
    pi = list(turn(pi))
    lit += pret(pi)
    lis,pi = turn3(pi)
    lit += lis
    pi = turnover(pi)
    lit += pret(pi)
    lis,pi = turn3(pi)
    lit += lis
    pi = list(shift(pi))
    pi = list(turn(pi))
    lit += pret(pi)
    lis,pi = turn3(pi)
    lit += lis
    pi = turnover(pi)
    lit += pret(pi)
    lis,pi = turn3(pi)
    lit += lis
    return sorted(lit)
def same(dice,dice2):
    if sorted(check(dice)) == sorted(check(dice2)):
        return True
    return False
def shift(dice):
    rem = dice[0][1]
    dice[0] = dice[0].replace(dice[0][1],dice[1][1])
    dice[1] = dice[1].replace(dice[1][1],dice[2][1])
    dice[2] = dice[2].replace(dice[2][1],dice[3][1])
    dice[3] = dice[3].replace(dice[3][1],rem)
    return dice
def turn(dice):
    remem = dice[0][2]
    dice[0] = dice[2][1]+dice[0][0]+dice[0][1]+".."
    dice[2] = "."+remem+"..."
    return dice
num = int(input())
li = []
for x in range(num):
    li2 = []
    for x in range(5):
        li2.append(input())
    space = input()
    li.append(li2)
lis = []
lis2 = []
for i,dice in enumerate(li):
    t = fit(force(dice))
    lis.append(check(t))
for i,x in enumerate(lis):
    io = []
    for u,y in enumerate(lis):
        if i == u:
            continue
        if y == x:
            io.append(u+1)
    if len(io) == 0:
        print("-")
    else:
        print(" ".join(map(str,io)))

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

input
20
.6...
.4...
31...
.25..
...

correct output
3 7 11 16
6
1 7 11 16
5 19
4 19
...

user output
3 7 11 16
6
1 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: ACCEPTED

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

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

Test 30

Group: 2

Verdict: ACCEPTED

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

Test 31

Group: 1, 2

Verdict: ACCEPTED

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

correct output
-
-

user output
-
-