天梯赛L2-003 月饼满分(笔记)
一、代码
月饼是中国人在中秋佳节时吃的一种传统食品,不同地区有许多不同风味的月饼。现给定所有种类月饼的库存量、总售价、以及市场的最大需求量,请你计算可以获得的最大收益是多少。
注意:销售时允许取出一部分库存。样例给出的情形是这样的:假如我们有 3 种月饼,其库存量分别为 18、15、10 万吨,总售价分别为 75、72、45 亿元。如果市场的最大需求量只有 20 万吨,那么我们最大收益策略应该是卖出全部 15 万吨第 2 种月饼、以及 5 万吨第 3 种月饼,获得 72 + 45/2 = 94.5(亿元)。
输入格式:
每个输入包含一个测试用例。每个测试用例先给出一个不超过 1000 的正整数 N 表示月饼的种类数、以及不超过 500(以万吨为单位)的正整数 D 表示市场最大需求量。随后一行给出 N 个正数表示每种月饼的库存量(以万吨为单位);最后一行给出 N 个正数表示每种月饼的总售价(以亿元为单位)。数字间以空格分隔。
输出格式:
对每组测试用例,在一行中输出最大收益,以亿元为单位并精确到小数点后 2 位。
输入样例:
3 20
18 15 10
75 72 45
输出样例:
94.50
二、思路
利用Collections对单价进行自定义排序,节点类需要继承Comparable
三、实现代码
import java.util.*;
import java.io.*;
public class Main {
public static void main(String[] args) throws IOException{
//Scanner sc=new Scanner(System.in);
Read sc=new Read();
int n=sc.nextInt();
int d=sc.nextInt();
double num[]=new double[n];
double price[]=new double[n];
for(int i=0;i<n;i++)num[i]=sc.nextDouble();
for(int i=0;i<n;i++)price[i]=sc.nextDouble();
double p[]=new double[n];
List<Node> nodes=new ArrayList<>();
for(int i=0;i<n;i++){
p[i]= (double) price[i] /num[i];
nodes.add(new Node(num[i],price[i],p[i]));
}
Collections.sort(nodes);
double ans=0;
for(int i=0;i<nodes.size();i++){
if(d>=nodes.get(i).num){
ans+=nodes.get(i).price;
d-=nodes.get(i).num;
}else{
ans+=(double) nodes.get(i).price/((double) nodes.get(i).num /(double) d);
break;
}
}
System.out.printf("%.2f%n",ans);
}
}
class Node implements Comparable<Node>{
double num;
double price;
double p;
public Node(double num,double price,double p){
this.num=num;
this.price=price;
this.p=p;
}
@Override
public int compareTo(Node o) {
return Double.compare(o.p,this.p);
}
}
class Read {
BufferedReader bfr = new BufferedReader(new InputStreamReader(System.in));
StreamTokenizer st = new StreamTokenizer(bfr);
public int nextInt() throws IOException {
st.nextToken();
return (int) st.nval;
}
public Double nextDouble() throws IOException {
st.nextToken();
return (Double) st.nval;
}
public String nextLine() throws IOException {
return bfr.readLine();
}
}