滴滴2017校招笔试编程题：餐馆（贪心+二分）Java

题目描述

某餐馆有n张桌子，每张桌子有一个参数：a 可容纳的最大人数； 有m批客人，每批客人有两个参数:b人数，c预计消费金额。 在不允许拼桌的情况下，请实现一个算法选择其中一部分客人，使得总预计消费金额最大

输入描述:
输入包括m+2行。 第一行两个整数n(1 <= n <= 50000),m(1 <= m <= 50000) 第二行为n个参数a,即每个桌子可容纳的最大人数,以空格分隔,范围均在32位int范围内。 接下来m行，每行两个参数b,c。分别表示第i批客人的人数和预计消费金额,以空格分隔,范围均在32位int范围内。

输出描述:
输出一个整数,表示最大的总预计消费金额

输入例子1:
3 5 2 4 2 1 3 3 5 3 7 5 9 1 10

输出例子1:
20

思路

先对桌子容量进行从小到大排序，再对m批客人的消费金额进行从大到小排序
用一个布尔数组作为标志变量：标志某个桌子是否已经被占用

贪心法：每次找到消费金额最多的批次的客人，再去验证桌子容量是否能容纳，若能容纳，选择容量刚好能满足的最小的桌子

二分查找：选定消费金额最多的客人后，找到合适的桌子容量（比客人数多的最小桌子）

完整实现

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[] N=new int[n];
        for(int i=0;i<n;i++){
            N[i]=sc.nextInt();    //N[i]=a
        }
        Arrays.sort(N);   //将桌子能容纳人数从小到大排序
        int[][] M=new int[m][2];
        for(int i=0;i<m;i++){
            for(int j=0;j<2;j++){
                M[i][j]=sc.nextInt();
            }
        }
        //将消费金额从大到小排序
        Arrays.sort(M, new Comparator<int[]>() {
            @Override
            public int compare(int[] a, int[] b) {
                return b[1]-a[1];
            }
        });
        
        boolean[] vis=new boolean[n];   //标记每张桌子是否被占
        
        long max=0L;   //最终结果————本来用的int，只通过50%，所以改成了long类型
        
        //遍历m批人
        for(int i=0;i<m;i++){
            //首先确保人数比桌子容量小,若人数超过最大桌子的容量，直接跳出本次循环
            if(M[i][0]>N[n-1]){
                continue;
            }
            int b=M[i][0];   //人数
            int c=M[i][1];  //金额
            int index=search(b,N);  //当前人数在N中的位置,再查看这个桌子是否被占了
            while(index<n && vis[index]==true){
                index++;
            }
            if(index<n){
                max+=c;
                vis[index]=true;
            }
        }
        System.out.println(max);
        
    }
    
    
    //二分查找，找到num在N数组中的位置（num要小于等于找到的桌子容量）
    public static int search(int num,int[] N){
        int left=0;
        int right=N.length-1;
        while(left<=right){
            int mid=(left+right)/2;
            if(N[mid]<num){
                left=mid+1;
            }
            else{
                right=mid-1;
            }
        }
        return left;
    }
}

要点：
如何对二维数组的列进行排序？而且是从大到小?
参考博客 使用Arrays.sort对二维数组分别按行和列排序