题目描述
为了找到自己满意的工作,牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下,牛牛选择报酬最高的工作。在牛牛选定了自己的工作后,牛牛的小伙伴们来找牛牛帮忙选工作,牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了,于是他只好把这个任务交给了你。
输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数,分别表示工作的数量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
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int m=sc.nextInt();
int[][] arr=new int[n][2];
int[] abi=new int[m];
TreeMap<Integer,Integer> map=new TreeMap<>();
HashMap<Integer,Integer> hs=new HashMap<>();
for(int i=0;i<n;i++){
arr[i][0]=sc.nextInt();
arr[i][1]=sc.nextInt();
}
for (int i=0;i<arr.length;i++){
hs.put(arr[i][0],arr[i][1]);
}
//排序
int[] arr1=new int[n];
for(int i=0;i<n;i++){
arr1[i]=arr[i][0];
}
Arrays.sort(arr1);
for(int i=0;i<n;i++){
arr[i][0]=arr1[i];
arr[i][1]=hs.get(arr[i][0]);
}
for(int i = 1; i < arr.length; i++) {
arr[i][1] = Math.max(arr[i-1][1], arr[i][1]);
}
for (int i=0;i<arr.length;i++){
map.put(arr[i][0],arr[i][1]);
}
for(int i=0;i<m;i++){
abi[i]=sc.nextInt();
Integer index=map.floorKey(abi[i]);
if (index!=null){
System.out.println(map.get(index));
}else{
System.out.println(0);
}
}
}
}
继续昨晚没通过的那道题,这次过了,注意是因为测试用例中存在难度与薪资不成正比的的情况,要全面考虑。
不过此代码空间时间都稍大,继续对其优化。