题目描述
某餐馆有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对二维数组分别按行和列排序