58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
import sys
|
|
|
|
ROCK = "rock"
|
|
SPICE = "spice"
|
|
|
|
|
|
def main():
|
|
lines = []
|
|
for line in sys.stdin:
|
|
lines.append(line.rstrip("\n"))
|
|
|
|
number_of_rocks, number_of_spices_types, capacity = map(int, lines[0].split())
|
|
rocks = []
|
|
spices = []
|
|
for line in lines[1 : number_of_rocks + 1]:
|
|
value, weight = map(int, line.split())
|
|
rocks.append(
|
|
{
|
|
"type": ROCK,
|
|
"total_value": value,
|
|
"weight": weight,
|
|
"cost_per_gram": value / weight,
|
|
}
|
|
)
|
|
for line in lines[number_of_rocks + 1 :]:
|
|
cost_per_gram, weight = map(int, line.split())
|
|
spices.append(
|
|
{
|
|
"type": SPICE,
|
|
"total_value": cost_per_gram * weight,
|
|
"weight": weight,
|
|
"cost_per_gram": cost_per_gram,
|
|
}
|
|
)
|
|
items = []
|
|
items.extend(rocks)
|
|
items.extend(spices)
|
|
items.sort(key=lambda x: x["cost_per_gram"], reverse=True)
|
|
|
|
cash_prize = 0
|
|
for item in items:
|
|
if item["type"] == ROCK and capacity - item["weight"] >= 0:
|
|
cash_prize += item["total_value"]
|
|
capacity -= item["weight"]
|
|
elif item["type"] == SPICE:
|
|
taking = min(capacity, item["weight"])
|
|
cash_prize += item["cost_per_gram"] * taking
|
|
capacity -= taking
|
|
if capacity == 0:
|
|
print(cash_prize)
|
|
return
|
|
print(cash_prize)
|
|
return
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|