結果

問題 No.1045 直方体大学
ユーザー pekempey
提出日時 2020-05-01 22:37:23
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,369 bytes
コンパイル時間 349 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 100,640 KB
最終ジャッジ日時 2024-12-25 13:25:00
合計ジャッジ時間 39,020 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 7 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

def input_int():
    return int(input())
def input_ints():
    return list(map(int, input().split()))
import numpy as np
N = input_int()
A = []
B = []
C = []
for _ in range(N):
    a, b, c = sorted(input_ints())
    A.append(a)
    B.append(b)
    C.append(c)
# 0: AB
# 1: AC
# 2: BC
def height(k):
    i, j = divmod(k, 3)
    if j == 0:
        return C[i]
    elif j == 1:
        return B[i]
    else:
        return A[i]
def face(k):
    i, j = divmod(k, 3)
    if j == 0:
        return A[i], B[i]
    elif j == 1:
        return A[i], C[i]
    else:
        return B[i], C[i]
INF = 10**9
dp = np.full((1 << N, N * 3), -INF, dtype=np.int32)
for i in range(N):
    dp[1 << i][i * 3 + 0] = height(i * 3 + 0)
    dp[1 << i][i * 3 + 1] = height(i * 3 + 1)
    dp[1 << i][i * 3 + 2] = height(i * 3 + 2)
def can(x, y):
    f1 = face(x)
    f2 = face(y)
    return f1[0] <= f2[0] and f1[1] <= f2[1]
g = [[can(i, j) for j in range(N * 3)] for i in range(N * 3)]
ans = 0
for i in range(1 << N):
    for j in range(N * 3):
        if dp[i][j] == -INF:
            continue
        ans = max(ans, dp[i][j])
        for kx in range(N):
            if i >> kx & 1: continue
            for ky in range(3):
                k = kx * 3 + ky
                if g[j][k]:
                    dp[i | 1 << kx][k] = max(dp[i | 1 << kx][k], dp[i][j] + height(k))
print(ans)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0