滴滴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 }