滴滴2016笔试:餐馆问题

某餐馆有n张桌子,每张桌子有一个参数:a可容纳的最大人数;有m批客人,每批客人有两个参数:b人数、c预计消费金额。在不允许拼桌的情况下,请实现一个算法选择其中一部分客人,使得总预计消费金额最大。

输入描述
输入包括m+2行。
第一行包括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


算法思路:贪心法解,把桌子按大小由小到大排序,把顾客批次按价格由大到小排序。然后顺次遍历桌子,每次选定桌子后顺次遍历顾客批次,可以的就坐进当前桌子。
C++程序:
 1 #include <iostream>
 2 #include <algorithm>
 3 #include <vector>
 4 using namespace std;
 5 
 6 struct person
 7 {
 8     int num;
 9     int price;
10 };
11 
12 bool cmp(person p1, person p2)
13 {
14     return p1.price>p2.price;
15 }
16 
17 int main()
18 {
19     int n,m;
20     vector<person> pp;
21 
22     cin>>n>>m;
23     vector<int> table;
24     for(int i=0; i<n; i++)
25     {
26         int tab;
27         cin>>tab;
28         table.push_back(tab);
29     }
30     for(int i=0; i<m; i++)
31     {
32         person tmp_p;
33         cin>>tmp_p.num>>tmp_p.price;
34         pp.push_back(tmp_p);
35     }
36 
37     sort(pp.begin(), pp.end(), cmp); // from big 2 small
38     sort(table.begin(), table.end()); // from small 2 big
39 
40     int total_price=0;
41     for(int i=0; i<table.size(); i++) // from small 2 big
42     {
43         for(int j=0; j<pp.size(); j++) // from big 2 small
44         {
45             if(pp[j].num <= table[i])
46             {
47                 total_price += pp[j].price;
48                 pp.erase(pp.begin()+j);
49                 break;
50             }
51         }
52     }
53 
54     cout<<total_price<<endl;
55 
56     return 0;
57 }

 

 

 






posted @ 2017-08-22 16:18  hedgehog小子  阅读(674)  评论(0编辑  收藏  举报