牛客网 餐馆

牛客网 餐馆

题目描述
某餐馆有n张桌子，每张桌子有一个参数：a 可容纳的最大人数； 有m批客人，每批客人有两个参数:b人数，c预计消费金额。 在不允许拼桌的情况下，请实现一个算法选择其中一部分客人，使得总预计消费金额最大
输入描述:
输入包括m+2行。 第一行两个整数n(1 <= n <= 50000),m(1 <= m <= 50000) 第二行为n个参数a,即每个桌子可容纳的最大人数,以空格分隔,范围均在32位int范围内。 接下来m行，每行两个参数b,c。分别表示第i批客人的人数和预计消费金额,以空格分隔,范围均在32位int范围内。
输出描述:
输出一个整数,表示最大的总预计消费金额

分析：使用贪心算法，将桌子容量A按从小到大排序，将客人B按消费金额从大到小排序。然后对客人进行一趟遍历，依次找出符合条件的客人，并从A中删除对应的桌子容量，最后可得出结果。注意，由于时间限制，在A中查找B中元素对应的位置时，应使用二分查找。

代码如下：

import bisect

n, m = map(int, raw_input().split())
A = map(int, raw_input().split())
B = []
for i in range(m):
    a, b = map(int, raw_input().split())
    B.append([a,b])

A.sort()
B.sort(key = lambda c:c[1], reverse = True)

maxCost = 0
for i in range(len(B)):
    j = bisect.bisect_left(A, B[i][0])
    if(j < 0 or j >= len(A)):
        continue
    del A[j]
    maxCost += B[i][1]

print maxCost