提出 #4936994


ソースコード 拡げる

X, Y, Z, K = map(int, input().split(' '))
A = sorted(list(map(int, input().split(' '))), reverse=True)
B = sorted(list(map(int, input().split(' '))), reverse=True)
C = sorted(list(map(int, input().split(' '))), reverse=True)

import heapq

# 合計値, a_idx, b_idx, c_idx
queue = [(-(A[0] + B[0] + C[0]), 0, 0, 0)]
# heapqueueにqueueをかえる
heapq.heapify(queue)

visited = set()
i = 0
for i in range(K):
    while True:
        v, a, b, c = heapq.heappop(queue)  # 候補の中から1番目が一番小さいtupleを取り出す
        if (a, b, c) not in visited:
            break
    visited.add((a, b, c))
    print(-v)

    if a < X - 1:
        heapq.heappush(queue, (-(A[a + 1] + B[b] + C[c]), a + 1, b, c))
    if b < Y - 1:
        heapq.heappush(queue, (-(A[a] + B[b + 1] + C[c]), a, b + 1, c))
    if c < Z - 1:
        heapq.heappush(queue, (-(A[a] + B[b] + C[c + 1]), a, b, c + 1))

提出情報

提出日時
問題 D - Cake 123
ユーザ cocodrips
言語 Python (3.4.3)
得点 400
コード長 931 Byte
結果 AC
実行時間 39 ms
メモリ 4212 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 400 / 400
結果
AC × 3
AC × 26
セット名 テストケース
Sample s1.txt, s2.txt, s3.txt
All in01.txt, in02.txt, in03.txt, in04.txt, in05.txt, in06.txt, in07.txt, in08.txt, in09.txt, in10.txt, in11.txt, in12.txt, in13.txt, in14.txt, in15.txt, in16.txt, in17.txt, in18.txt, in19.txt, in20.txt, in21.txt, in22.txt, in23.txt, s1.txt, s2.txt, s3.txt
ケース名 結果 実行時間 メモリ
in01.txt AC 38 ms 3804 KiB
in02.txt AC 38 ms 3828 KiB
in03.txt AC 37 ms 3828 KiB
in04.txt AC 37 ms 3784 KiB
in05.txt AC 37 ms 3804 KiB
in06.txt AC 37 ms 3800 KiB
in07.txt AC 38 ms 3956 KiB
in08.txt AC 39 ms 3956 KiB
in09.txt AC 37 ms 3920 KiB
in10.txt AC 35 ms 3700 KiB
in11.txt AC 35 ms 3700 KiB
in12.txt AC 36 ms 3700 KiB
in13.txt AC 37 ms 4084 KiB
in14.txt AC 36 ms 4084 KiB
in15.txt AC 36 ms 4212 KiB
in16.txt AC 18 ms 3064 KiB
in17.txt AC 18 ms 3064 KiB
in18.txt AC 36 ms 4212 KiB
in19.txt AC 36 ms 4212 KiB
in20.txt AC 36 ms 4212 KiB
in21.txt AC 37 ms 4052 KiB
in22.txt AC 37 ms 4212 KiB
in23.txt AC 36 ms 4212 KiB
s1.txt AC 18 ms 3064 KiB
s2.txt AC 18 ms 3064 KiB
s3.txt AC 18 ms 3064 KiB