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

来源:互联网 发布:js数学公式编辑器 编辑:程序博客网 时间:2024/05/16 05:11

题目描述

某餐馆有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-8import sysimport bisectwhile 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

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

原创粉丝点击