标题:餐馆 | 时间限制:1秒 | 内存限制:65536K
某餐馆有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范围内。
输出描述:
输出一个整数,表示最大的总预计消费金额.
示例1:
输入
3 5
2 4 2
1 3
3 5
3 7
5 9
1 10
输出
20
代码:
#include <iostream>
#include <string>
#include <memory>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
// 每批次顾客的人数和预消费金额
struct SCustomer
{
typedef shared_ptr<SCustomer> Ptr;
uint32_t nNumber; // 顾客人数
uint32_t nMoney; // 消费金额
};
// 预算金额降序排列,金额相同消费人数升序排列
int CompareOder(SCustomer::Ptr A, SCustomer::Ptr B)
{
if (A->nMoney == B->nMoney)
{
return A->nNumber < B->nNumber;
}
return A->nMoney > B->nMoney;
}
int main(int argc, char* argv[])
{
uint32_t nTable; // 餐桌数目
uint32_t nCounts; // 顾客批次数
long long nAllMoney = 0;
multimap<int, int> TableMap;
vector<SCustomer::Ptr> CustomerVec;
// 初始餐桌数目和批次数
cin >> nTable;
cin >> nCounts;
// 初始化每张餐桌就餐人数
for (int i = 0; i < nTable; ++i)
{
uint32_t nSize;
cin >> nSize;
TableMap.insert({ nSize, 1 });
}
// 初始化每批次人数和预消费金额
for (int i = 0; i < nCounts; i++)
{
uint32_t x, y;
cin >> x;
cin >> y;
SCustomer::Ptr pCustomer = make_shared<SCustomer>();
pCustomer->nNumber = x;
pCustomer->nMoney = y;
CustomerVec.push_back(pCustomer);
}
// 排序
sort(CustomerVec.begin(), CustomerVec.end(), CompareOder);
// 计算总预算
for (int i = 0; i < nCounts; i++)
{
auto iter = TableMap.lower_bound(CustomerVec[i]->nNumber);
if (iter != TableMap.end())
{
TableMap.erase(iter);
nAllMoney += CustomerVec[i]->nMoney;
}
}
std::cout << nAllMoney << std::endl;
system("pause");
return 0;
}
测试图: