题目描述
为了找到自己满意的工作,牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下,牛牛选择报酬最高的工作。在牛牛选定了自己的工作后,牛牛的小伙伴们来找牛牛帮忙选工作,牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了,于是他只好把这个任务交给了你。
输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数,分别表示工作的数量N(N<=100000)和小伙伴的数量M(M<=100000)。
接下来的N行每行包含两个正整数,分别表示该项工作的难度Di(Di<=1000000000)和报酬Pi(Pi<=1000000000)。
接下来的一行包含M个正整数,分别表示M个小伙伴的能力值Ai(Ai<=1000000000)。
保证不存在两项工作的报酬相同。
输出描述:
对于每个小伙伴,在单独的一行输出一个正整数表示他能得到的最高报酬。一个工作可以被多个人选择。
示例:
输入
3 3
1 100
10 1000
1000000000 1001
9 10 1000000000
输出
100
1000
1001
分析:这道题目不能直接暴力遍历,因为测试用例数据量特别大,很容易超时,所以需要时间复杂度更低的算法。
正确解题思路:根据工作的报酬由高到低对N项工作进行排序,从报酬最高的工作开始选(这里体现了贪心算法),根据小伙伴的能力对M个小伙伴进行排序,排序前需要给每个小伙伴进行编号以便最后按原先的顺序输出。每个小伙伴根据报酬由高到低选工作,当小伙伴的能力大于或等于当前工作难度的时候,就帮小伙伴找到了最合适的工作。重点来了,因为下一个小伙伴的能力肯定小于或等于上一个小伙伴的能力,所以这个小伙伴就可以从当前工作继续向下查找。帮每个小伙伴找到工作后,最后根据原先编号的序号排序后再输出。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
typedef long T;
bool cmp1(vector<T> a, vector<T> b) {//谓词,按升序排
return a[0] < b[0];
}
bool cmp2(vector<T> a, vector<T> b) {//谓词,按降序排
return a[1] > b[1];
}
int main()
{
vector<vector<T>> work;//二维数组,存储工作信息
vector<vector<T>> peo;//二维数组,存储小伙伴信息
int n, m;
cin >> n >> m;
while (n--) {
T di, pi;
cin >> di >> pi;
work.push_back({di, pi});
}
sort(work.begin(), work.end(), cmp2);//将工作根据报酬由高到低排序
int k = 0;//存储小伙伴的编号
while (m--) {
T ai;
cin >> ai;
peo.push_back({k++, ai});
}
sort(peo.begin(), peo.end(), cmp2);//将小伙伴根据能力由高到低排序
int j = 0;
for (int i = 0; i < peo.size(); i++) {
while (j < work.size() && peo[i][1] < work[j][0]) ++j;
if (j == work.size())//将当前小伙伴的能力修改成所选工作的报酬
peo[i][1] = 0;
else
peo[i][1] = work[j][1];
}
sort(peo.begin(), peo.end(), cmp1);
for (int i = 0; i < peo.size(); i++)//将小伙伴根据编号排序
cout << peo[i][1] << endl;
return 0;
}