餐馆牛客网【贪心+二分】

热度指数：24039 时间限制：1秒空间限制：65536K

某餐馆有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

输出

题解：需要消费金额最大，我们根据消费金额大小排序，不用考虑是否需要按金额除以那一桌得到人数，因为桌子大小不固定的，例如：

人数消费金额

35 10

10 10

6 11

2 10

我们只需按金额排序，然后再查找满足人数的桌子中最少人数的桌子，查找时使用二分，不然会超时。

注意：桌子可容纳的最大人数可能会是一样的，例如：1 2 2 2 3 3 3 5

所以在二分的时候，我们要找到最左边的那个重复的，如果找到其他的，是不行的，如果二分没有找到相应的桌子，就需要返回比他大一点的桌子。

例如人数是4，在二分是要返回5的下标。

import java.util.Arrays;
import java.util.Comparator;
import java.util.Scanner;
public class Test07 {
    public static void main(String[] args) {
        Scanner in=new Scanner(System.in);
        while(in.hasNext()){
            int n=in.nextInt();
            int m=in.nextInt();
            long max=0L;

            int[] a=new int[n];
            int[] vis=new int[n];    //标记，查看这个桌子是否被用了，如果是就置为1
            int[][] num=new int[m][2];
            for (int i = 0; i < n; i++) {//每个桌承受人数
                a[i]=in.nextInt();
            }
            Arrays.sort(a);

            for (int i = 0; i < m; i++) {
                num[i][0]=in.nextInt();
                num[i][1]=in.nextInt();
            }

            Arrays.sort(num, new Comparator<int[]>() {
                @Override
                public int compare(int[] a, int[] b) {
                    return b[1]-a[1];
                }
            });

            for (int i = 0; i < m; i++) {   //遍历m批人
                if(a[n-1]<num[i][0]){
                    continue;
                }
                int nu=num[i][0];       //人数
                int price=num[i][1];    //价值

                //找到nu在n的位置
                int index=dic(nu,a);
                while(index<n && vis[index]==1){   //向后找到没有用的桌子
                    index++;
                }
                if(index<n){
                    max+=price;
                    vis[index]=1;
                }
            }
            System.out.println(max);

        }
    }
    public static int dic(int num,int[] a){
        int left=0;
        int right=a.length-1;
        int mid=0;
        while(left<=right){
            mid=(right+left)>>1;  //二分
            if(num<=a[mid]){   //这个要小于等于才能找到最左边的，例如找到了num==a[mid],那么还
                right=mid-1;   //继续二分，继续找左边的，直到找到最坐标的
            }else{
                left=mid+1;
            }
        }
        return left;
      }
}

餐馆 牛客网 【贪心+二分】

输入

输出

猜你喜欢

餐馆牛客网【贪心+二分】