結果

問題 No.1045 直方体大学
ユーザー pekempeypekempey
提出日時 2020-05-01 22:37:23
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,369 bytes
コンパイル時間 120 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 63,480 KB
最終ジャッジ日時 2024-06-07 10:58:26
合計ジャッジ時間 9,518 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 537 ms
51,256 KB
testcase_01 AC 540 ms
44,820 KB
testcase_02 AC 526 ms
44,688 KB
testcase_03 AC 536 ms
44,688 KB
testcase_04 AC 531 ms
44,404 KB
testcase_05 AC 626 ms
44,300 KB
testcase_06 AC 616 ms
44,312 KB
testcase_07 AC 615 ms
44,688 KB
testcase_08 TLE -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
権限があれば一括ダウンロードができます

ソースコード

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)
0