結果

問題 No.2852 Yakitori Optimization Problem
ユーザー Yuyo1984Yuyo1984
提出日時 2024-08-25 16:18:45
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
WA  
実行時間 -
コード長 766 bytes
コンパイル時間 778 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 97,616 KB
最終ジャッジ日時 2024-08-25 16:19:07
合計ジャッジ時間 20,965 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 AC 1,475 ms
97,616 KB
testcase_16 AC 1,464 ms
97,336 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

n, k = map(int, input().split())
A = [*map(int, input().split())]
B = [*map(int, input().split())]
C = [*map(int, input().split())]

ranking = list()
for i in range(n):
    ranking.append((A[i] + B[i], 0, i))
    ranking.append((A[i] + C[i], 1, i))

used = set()

ans = 0
ranking.sort(reverse=True)

made_salt = k
made_sauce = n - k
for i in range(len(ranking)):
    value = ranking[i][0]
    taste = ranking[i][1]
    index = ranking[i][2]
    if taste == 0 and made_salt > 0 and index not in used:
        ans += value
        made_salt -= 1
        used.add(index)
    if taste == 1 and made_sauce > 0 and index not in used:
        ans += value
        made_sauce -= 1
        used.add(index)
    if made_salt <= 0 and made_sauce <= 0:
        break

print(ans)
0