网易笔试题 牛牛找工作

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ymybxx/article/details/79968951

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

一开始看到题,首先想到的肯定是穷举,先把工作按照工资从高到低排序 然后对每个员工的能力从工作第一个开始比较,找到能力足够的就输出,但是时间超时了

只能进行优化了.先将工作按照难度从小到大排序,对每个难度进行更新,可以得出每个难度对应能得到最高的工资是多少,将该信息存入到TreeMap中,再利用TreeMap的floorKey输出结果

public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        //工作数量
        int workNum = scanner.nextInt();
        //伙伴数量
        int pNum = scanner.nextInt();
        int[][] arr = new int[workNum][2];
        for (int i = 0; i < workNum ; i++) {
            int ai = scanner.nextInt();
            int sa = scanner.nextInt();
            arr[i][0] = ai;
            arr[i][1] = sa;
        }
        Arrays.sort(arr, new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return o1[0] - o2[0];
            }
        });
        for (int i = 1; i < workNum ; i++) {
            arr[i][1] = Math.max(arr[i - 1][1],arr[i][1]);
        }
        TreeMap<Integer, Integer> map = new TreeMap<>();
        for (int i = 0; i < workNum; i++) {
            map.put(arr[i][0],arr[i][1]);
        }
        for (int i = 0; i < pNum; i++) {
            int ai = scanner.nextInt();
            Integer index = (Integer)map.floorKey(ai);
            if(index != null) System.out.println(map.get(index));
            else System.out.println(0);

        }
    }
阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页