昨天刷网易新出的校招题目,第一题,就遇到个硬茬。
时间限制:2秒
空间限制:65536K
为了找到自己满意的工作,牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下,牛牛选择报酬最高的工作。在牛牛选定了自己的工作后,牛牛的小伙伴们来找牛牛帮忙选工作,牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了,于是他只好把这个任务交给了你。
输入描述:
每个输入包含一个测试用例。 每个测试用例的第一行包含两个正整数,分别表示工作的数量N(N<=100000)和小伙伴的数量M(M<=100000)。 接下来的N行每行包含两个正整数,分别表示该项工作的难度Di(Di<=1000000000)和报酬Pi(Pi<=1000000000)。 接下来的一行包含M个正整数,分别表示M个小伙伴的能力值Ai(Ai<=1000000000)。 保证不存在两项工作的报酬相同。
输出描述:
对于每个小伙伴,在单独的一行输出一个正整数表示他能得到的最高报酬。一个工作可以被多个人选择。
输入例子1:
3 3 1 100 10 1000 1000000000 1001 9 10 1000000000
输出例子1:
100 1000 1001
很显然,这种题目,直接比较着做是不可能全部通过的。
我最开始想的是用二分查找:
- 把所有工作按难度排序
- 找到每个朋友所能胜任的难度最大的工作(二分法搜索 速度快)
- 从这个工作倒着往回找所有可以从事的工作中报酬最大的那一个,输出
万万没想到,这题目比想象的硬多了,居然还是只通过了40%。
想了一晚上,没想出啥办法,看了一下讨论区,才知道要用map来做。
map是c++里的一个容器类,可以存放键值对,因为是用红黑树存的,所以存放在里面的键值对自动按照键(key)的大小排序。
要注意的是,所有的key都必须是独一无二的,不能重复,如果重复插入key相等的键值对,则新值(value)会覆盖旧值。
了解map之后,这个题终于有了快速解出的可能。不多说,上代码:
#include <iostream>
#include <map>
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int n=0, m=0;
map<int, int> works;
vector<int> hard; //工作难度
vector<int> ablity; //各朋友的能力
int main(){
freopen("work.in", "r", stdin);
cin>>n>>m;
int D=0, P=0;
for(int i=0;i<n;++i){
cin>>D;
cin>>P;
works[D] = works[D]>P? works[D]:P; //如果有难度相同的工作,则用报酬值大的覆盖报酬值小的
hard.push_back(D);
}
int A;
for(int i=0;i<m;++i){
cin>>A;
hard.push_back(A);
ablity.push_back(A);
works[A] = works[A]>0? works[A]:0; //如果难度正好等于某人能力的工作不存在,则设这一难度对应的报酬为零。
}
sort(hard.begin(), hard.end()); //对所有key排序,包括工作难度和朋友的能力
int max=0;
//通过遍历,将每个难度(Key)所对应的报酬(Value)都赋值为不超过这个难度的工作中报酬最大的
for(int i=0; i<m+n; ++i){
max = max>works[hard[i]]? max:works[hard[i]];
works[hard[i]] = max;
}
for(int i=0;i<m;++i){
cout<<works[ablity[i]]<<endl;
}
return 0;
}
思路在注释里写得很清楚了,不赘述。
参考:题目下方讨论区