結果

問題 No.2984 [Cherry Anniversary 4] 満開の願いを込めた 27 の桜
ユーザー 👑 Kazun
提出日時 2024-10-22 00:45:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,267 ms / 4,000 ms
コード長 1,595 bytes
コンパイル時間 343 ms
コンパイル使用メモリ 82,692 KB
実行使用メモリ 91,152 KB
最終ジャッジ日時 2024-12-08 23:30:42
合計ジャッジ時間 41,940 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import product, permutations

def matrix_input(h):
    cin = [None] * h
    for i in range(h):
        cin[i] = list(map(int, input().split()))
    return cin

def optimized_step(A):
    if A == (0, 1, 2):
        return 0
    elif A == (1, 0, 2) or A == (2, 1, 0) or A == (0, 2, 1):
        return 1
    else:
        return 2

def solve():
    N = 3

    A = [None] * N
    for i in range(N):
        A[i] = matrix_input(N)

    X = matrix_input(N)
    Y = matrix_input(N)
    Z = matrix_input(N)

    def check(B):
        for j in range(N):
            for k in range(N):
                if B[0][j][k] + B[1][j][k] + B[2][j][k] != X[j][k]:
                    return False

        for i in range(N):
            for k in range(N):
                if B[i][0][k] + B[i][1][k] + B[i][2][k] != Y[i][k]:
                    return False

        for i in range(N):
            for j in range(N):
                if B[i][j][0] + B[i][j][1] + B[i][j][2] != Z[i][j]:
                    return False

        return True

    inf = float('inf')
    ans = inf
    for alpha, beta, gamma in product(permutations(range(N)), repeat = 3):
        B = [[[A[alpha[i]][beta[j]][gamma[k]] for k in range(N)] for j in range(N)] for i in range(N)]

        if check(B):
            ans = min(ans, optimized_step(alpha) + optimized_step(beta) + optimized_step(gamma))

    return ans if ans < inf else -1

#==================================================
import sys
input = sys.stdin.readline
write = sys.stdout.write

T = int(input())
write("\n".join(map(str, [solve() for _ in range(T)])))
0