为了找到自己满意的工作,牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下,牛牛选择报酬最高的工作。在牛牛选定了自己的工作后,牛牛的小伙伴们来找牛牛帮忙选工作,牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了,于是他只好把这个任务交给了你。
输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数,分别表示工作的数量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
输出
100
1000
1001
解题思路:
思路比较清晰,先按照难度排序,然后更新每个工作的报酬为当前难度所能获取的最大报酬,最后二分查找确定能胜任的最大工作难度及其最大报酬。时间复杂度O(nlogn)
需要注意二分查找是查找小于等于key的位置,需要对普通二分查找更改,。
需要考虑到不能胜任任何工作的情况!
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
int jobNum = sc.nextInt();
int peopleNum = sc.nextInt();
Job[] jobs = new Job[jobNum];
for(int i=0; i<jobNum; i++){
jobs[i] = new Job(sc.nextInt(), sc.nextInt());
}
int[] abilitys = new int[peopleNum];
for(int i=0; i<peopleNum; i++)
abilitys[i] = sc.nextInt();
int[] res = new int[peopleNum];
helper(jobs, abilitys, res);
for(int i=0; i<peopleNum; i++)
System.out.println(res[i]);
}
private static void helper(Job[] jobs, int[] abilitys, int[] res){
//按照难度排序
Arrays.sort(jobs, new Comparator<Job>(){
@Override
public int compare(Job a, Job b){
if(a.diff != b.diff)
return a.diff-b.diff;
else
return a.money-b.money;
}
});
//更新每个工作的报酬为当前难度所能获取的最大报酬
for(int i=1; i<jobs.length; i++){
if(jobs[i-1].money > jobs[i].money)
jobs[i].money = jobs[i-1].money;
}
//二分查找确定能胜任的最大工作难度及其最大报酬
for(int i=0; i<abilitys.length; i++){
int x = binarySearch(jobs, abilitys[i]);
if(x == -1)
res[i] = 0;
else
res[i] = jobs[x].money;
}
}
// 返回小于等于key的索引,如果没有,那么返回-1
static int binarySearch(Job[] array, int key) {
int l = 0;
int r = array.length - 1;
while (l <= r) {
int mid = l + ((r - l) >>> 1);
if (array[mid].diff > key)
r = mid - 1;
else
l = mid + 1;
}
return r;
}
static class Job{
public int diff;
public int money;
public Job(int diff, int money){
this.diff = diff;
this.money = money;
}
}
}
二分查找的版本二:防止出现mid与l在一边死循环,这里将mid = (l+r+1)/2
import java.util.*;
public class Main{
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
int jobNum = sc.nextInt();
int peopleNum = sc.nextInt();
Job[] jobs = new Job[jobNum];
for(int i=0; i<jobNum; i++){
jobs[i] = new Job(sc.nextInt(), sc.nextInt());
}
int[] abilitys = new int[peopleNum];
for(int i=0; i<peopleNum; i++)
abilitys[i] = sc.nextInt();
int[] res = new int[peopleNum];
helper(jobs, abilitys, res);
for(int i=0; i<peopleNum; i++)
System.out.println(res[i]);
}
private static void helper(Job[] jobs, int[] abilitys, int[] res){
//按照难度排序
Arrays.sort(jobs, new Comparator<Job>(){
@Override
public int compare(Job a, Job b){
if(a.diff != b.diff)
return a.diff-b.diff;
else
return a.money-b.money;
}
});
//更新每个工作的报酬为当前难度所能获取的最大报酬
for(int i=1; i<jobs.length; i++){
if(jobs[i-1].money > jobs[i].money)
jobs[i].money = jobs[i-1].money;
}
//二分查找确定能胜任的最大工作难度及其最大报酬
for(int i=0; i<abilitys.length; i++){
int x = binarySearch(jobs, abilitys[i]);
if(abilitys[i] < x)
res[i] = 0;
else
res[i] = jobs[x].money;
}
}
// 返回小于等于key的索引
static int binarySearch(Job[] array, int key) {
int l = 0;
int r = array.length - 1;
while (l < r) {
int mid = (r + l + 1) >>> 1;
if (array[mid].diff > key)
r = mid - 1;
else
l = mid;
}
return l;
}
static class Job{
public int diff;
public int money;
public Job(int diff, int money){
this.diff = diff;
this.money = money;
}
}
}