【网易真题】牛牛找工作

哎 一时兴起做了一套网易真题，发现不会的不止一点点

哎 赶紧学习吧

题目描述：

链接：https://www.nowcoder.com/questionTerminal/5e1b251c90ba4e6982cb349076ed4449
来源：牛客网
为了找到自己满意的工作，牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下，牛牛选择报酬最高的工作。在牛牛选定了自己的工作后，牛牛的小伙伴们来找牛牛帮忙选工作，牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了，于是他只好把这个任务交给了你。 
输入描述:
每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数，分别表示工作的数量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

分析：刚开始的思路是通过一个TreeMap来存输入的工作难度和工作报酬键值对，当遇到相同的工作难度的时候，就将当前难度下的最大报酬存进来，因为毕竟一个工作可以被多个人选择嘛，只要保存某一个难度下的最大报酬就好了。然后对于每个小伙伴，遍历容器中所有键小于等于其难度的值，找到最大值。思路很简单 但是超时了

这个思路里要注意遍历容器 Iterator的方法，在写代码的时候还是忘了又去现查的……

TreeMap<Integer,Integer> map=new TreeMap<Integer,Integer>();
Iterator<Map.Entry<Integer, Integer>> it = map.entrySet().iterator();
while(it.hasNext()){
                Map.Entry<Integer, Integer> entry = it.next();

                //entry.getKey()

                //entry.getValue()

 }

以下为借鉴了别人的思路的方法，不光是想记录一下解题的思路，一些代码实现的细节也需要巩固记忆！

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();
        TreeMap<Integer,Integer> map=new TreeMap<Integer,Integer>();
        int[][] work=new int[n][2];
  
        for(int i=0;i<n;i++){
            int temp1=sc.nextInt();
            int temp2=sc.nextInt();
            work[i][0]=temp1;
            work[i][1]=temp2;
            
        }
        //对work数组根据第一维度，即工作难度排序
        Arrays.sort(work,new Comparator<int[]>(){
                @Override
                public int compare(int[] num1,int []num2){
                    return num1[0]-num2[0];
                }
            });
        //动态规划思想 维护数组中保存当前工作难度下的最大收益
        //然后存到一个TreeMap中 
        //主要为了后面利用其floorKey函数找到不大于某一个值的最大键值
        map.put(work[0][0],work[0][1]);
        for(int k=1;k<n;k++){
            work[k][1]=Math.max(work[k][1],work[k-1][1]);
            map.put(work[k][0],work[k][1]);
        }
 
        for(int j=0;j<m;j++){
            int friend=sc.nextInt();
            Integer index=map.floorKey(friend);
            if(index==null){
                System.out.println(0);
            }else{
                System.out.println(map.get(index));
            }
                
            
        }
    }
}

重写Arrays.sort()方法实现按照自己的意愿对数组进行排序

看到一篇博客介绍Comparator讲的很清楚

TreeMap的floorKey( )方法，返回一个不大于该输入的最大键值，如果没有返回null，注意index要声明成Integer类型；

还有动态规划的思想 需要巩固