CODE THANKS FESTIVAL 2017 C - Factory

提出

code: python

import heapq

n, k = map(int, input().split())

a = []

b = []

for i in range(n):

a.append(A)

b.append(B)

# heapqからindex取れないと、何を足して格納すればいいかわからない

# あらかじめ足しておく？

# だとしたらsortすればいいわけでheapを使う必要はない

# time = []

# for i in range(n):

# for j in range(1000):

# time

解説

code: python

from heapq import heappop

from heapq import heappush

from heapq import heapify

n, k = map(int, input().split())

hq = list()

heapify(hq)

ans = 0

for i in range(n):

ai, bi = map(int, input().split())

# 操作前の所要時間とその要素のインデックスをセット

heappush(hq, (ai, i))

for i in range(k):

# 最小所要時間とそのときの要素のインデックスを取り出す

time, index = heappop(hq)

ans += time

# 操作後に所要時間を更新

print(ans)

テーマ

メモ