【Java】牛牛找工作

题目描述

为了找到自己满意的工作，牛牛收集了每种工作的难度和报酬。牛牛选工作的标准是在难度不超过自身能力值的情况下，牛牛选择报酬最高的工作。在牛牛选定了自己的工作后，牛牛的小伙伴们来找牛牛帮忙选工作，牛牛依然使用自己的标准来帮助小伙伴们。牛牛的小伙伴太多了，于是他只好把这个任务交给了你。

输入描述:

每个输入包含一个测试用例。
每个测试用例的第一行包含两个正整数，分别表示工作的数量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

package 牛客;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Scanner;
import java.util.TreeMap;

public class 牛牛找工作 {
    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=0;i<arr.length;i++){
            map.put(arr[i][0],arr[i][1]);
        }

        for(int i=0;i<m;i++){
            int abli=sc.nextInt();
            Integer index=map.floorKey(abli);
            if (index!=null){
                System.out.println(map.get(index));
            }else{
                System.out.println(0);
            }
        }

    }
}

为了简单易懂我使用了HashMap进行二维数组的存取排序，可能代码不够完善只通过了30%的测试用例。明天继续。