PTA D1-C卡路里 python

冰箱里有n种食物，第i种食物有mi​ cal(卡路里)，以及含有vi​营养物质。计算鸭制定了未来k天减肥计划，第j天最多摄入cj​ cal(卡路里)，并且每天只吃一种食物。

问：计算鸭可以摄入营养物质的最大值是多少。

输入格式:

输入的第一行给出两个整数n,k,含义如题意所示；

接下来n行，每行两个整数mi​,vi​,含义如题意所示；

接下来k行，每行一个整数cj​,含义如题意所示；

1≤n,k≤3×105

1≤mi​,vi​≤106

1≤cj​≤108

输出格式:

输出一个整数——表示计算鸭可以摄入的最多营养物质的含量。

输入样例:

3 2
1 65
5 23
2 99
10
2

输出样例:

164

import heapq

def max_nutrition(n, k, foods, calorie_limits):
    foods.sort()
    calorie_limits.sort()
    
    max_heap = []
    total_nutrition = 0
    food_index = 0
    

    for c in calorie_limits:
        while food_index < n and foods[food_index][0] <= c:
            heapq.heappush(max_heap, -foods[food_index][1])  # Use negative to simulate max heap