算法学习笔记9-贪心算法

题目描述

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

输出描述

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

#coding=utf-8
import sys
import bisect
while True:
    line=sys.stdin.readline().strip()
    if line:
        n,m=map(int,line.split())
        d=map(int,sys.stdin.readline().strip().split())
        c=[]
        for i in xrange(m):
            c.append(map(int,sys.stdin.readline().strip().split()))
        c.sort(key=lambda x:(x[1],x[0]),reverse=True)
        #生成一个二维数组且优先按照顾客预计消费金额降序排列
        d.sort()#按照每个桌子能容纳人数升序排列
        money=0
        for i in c:
            f=bisect.bisect_left(d,i[0])#二分查找坐得下的桌子
            if f!=len(d):
                money+=i[1]
                del d[f]#删除这张桌子，因为已经被预定
        print money
    else:
        break

算法的主要思想是：每次选择预计消费金额最大的客人，直到桌子用完。贪心算法的核心就是：每一步都选择当前最优的。