【牛客】餐馆（贪心+二分，带图解）

一、题目描述

某餐馆有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范围内。

三、输出描述

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

四、示例

五、思路分析

通过分析题目，和题中所给出的示例，我的具体分析如下图所示：

六、代码实现

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while(sc.hasNext()){
            int n = sc.nextInt();//n张桌子
            int m = sc.nextInt();//m批客人
            int[] a = new int[n];// a
            int[][] bc = new int[m][2];// b、c

            for (int i = 0; i < n; i++) {//每个桌可容纳人数
                a[i] = sc.nextInt();
            }
            Arrays.sort(a);//对桌子容纳人数从小到大排序

            for (int i = 0; i < m; i++) {//第i批客人的人数和预计消费金额
                bc[i][0] = sc.nextInt();
                bc[i][1] = sc.nextInt();
            }
            Arrays.sort(bc, new Comparator<int[]>() {//对消费金额进行排序
                @Override
                public int compare(int[] a, int[] b) {
                    return b[1]-a[1];
                }
            });

            long max = 0L;//最大总预计消费金额
            int[] isUsed = new int[n];//标记,桌子被用就置为1

            for (int i = 0; i < m; i++) {//遍历m批人
                if(a[n-1] < bc[i][0]){ //如果第i批人数比最大容纳量还大就跳过
                    continue;
                }
                int num = bc[i][0];      //第i批人数
                int price = bc[i][1];    //第i批消费金额

                //找到合适的桌子坐
                int index = binarySearch(num,a);
                while(index<n && isUsed[index]==1){
                    index++;
                }
                if(index<n){
                    max += price;
                    isUsed[index] = 1;
                }
            }
            System.out.println(max);

        }
    }
    //二分查找
    public static int binarySearch(int num, int[] a){
        int left = 0;
        int right = a.length-1;
        int mid = 0;
        while(left <= right){
            mid = (right+left)/2;
            if(num <= a[mid]){
                right = mid-1;
            }else{
                left = mid+1;
            }
        }
        return left;
    }
}