餐馆——滴滴面试

题目描述

某餐馆有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范围内。

输出描述:

输出一个整数,表示最大的总预计消费金额

输入	输出
3 5 2 4 2 1 3 3 5 3 7 5 9 1 10	20

思路：

将m批客人，按消费额由大到小排序，将餐桌按人数由小到大排序，将顾客放置到相应的桌上

代码1：

n,m=[int(x) for x in input().split()]
peoples=[int(x) for x in input().split()]
ps=sorted(peoples)
sells=[]
sell_sum=0
for i in range(m):
    sell=[int(x) for x in input().split()]
    sells.append(sell)
sells_sort=sorted(sells,key=lambda x:x[1],reverse=True)
for s in sells_sort:
    for p in ps:
        if s[0]<=p:
            sell_sum+=s[1]
            ps.remove(p)
            break
print(sell_sum)

代码2：

import bisect
n, m = [int(x) for x in input().split()]
a = [int(x) for x in input().split()]
A = []
for i in range(m):
    A.append([int(x) for x in input().split()])
a.sort()
A.sort(key=lambda x: x[1], reverse=True)
counter = 0
for i in A:
    index = bisect.bisect_left(a, i[0])
    if index != len(a):
        counter += i[1]
        del a[index]
print(counter)