很简单的贪心算法,不过一开始做的时候还是有好多情况没有考虑到
思路很简单,读取输入,然后计算各种月饼单价,按单价对月饼进行排序,优先选择单价高对月饼。
下面说说一开始犯的几个错误。(我是用java写的)
1:一开始提交,全是答案错误,于是检查题目,发现要求保留两位小数。于是使用DecimalFormat工具,保留两位小数,然后试试看。
DecimalFormat df = new DecimalFormat("#.00");
String result = df.format(amount);
2:再次提交,多了几个正确答案,但是还是有答案错误,想了一下,如果月饼需求为0,则应该输出"0.00",但是我的程序输出了".00"
3:再次提交,大部分都正确了,但是还是有一个答案错误,想了好久没有结果,后来上网查了一下,经过网友提示,还需要考虑供不应求对情况,即月饼的总量小于需求,这时应返回所有月饼的总价。于是我写了一个这种情况的测试用例,测试,发现明明是运行时错误。。。
好的,再改。
4:终于全部正确了
感想:在leetcode上做得习惯了,看到错的结果能对照测试用例有目的性地该,在PAT上真有点不太适应。
下面附上我的代码
import java.io.InputStreamReader;
import java.text.DecimalFormat;
import java.util.Arrays;
import java.io.BufferedReader;
import java.io.IOException;
public class Main
{
public static BufferedReader br =
new BufferedReader(new InputStreamReader(System.in));
public static void main(String[] args) throws IOException
{
String[] info = br.readLine().split(" ");
int kind = Integer.valueOf(info[0]);
double request = Integer.valueOf(info[1]);
String[] stockArr = br.readLine().split(" ");
String[] priceArr = br.readLine().split(" ");
MoonCake[] mc = new MoonCake[kind];
for(int i = 0; i < kind; i++)
{
double s = Double.valueOf(stockArr[i]);
double p = Double.valueOf(priceArr[i])/s;
MoonCake m = new MoonCake(s, p);
mc[i] = m;
}
double count = 0;
int index = 0;
Arrays.sort(mc);
while(request > 0)
{
if(mc[index].stock >= request )
{
count += mc[index].price * request;
request = 0;
}else
{
request -= mc[index].stock;
count += mc[index].price * mc[index].stock;
index+=1;
}
}
DecimalFormat df = new DecimalFormat("#.00");
String result = df.format(count);
if(result.equals(".00"))
{
System.out.println("0.00");
}
else {
System.out.println(result);
}
}
}
class MoonCake implements Comparable<MoonCake>
{
public double stock;
public double price;
public MoonCake(double stock, double price) {
// TODO Auto-generated constructor stub
this.stock = stock;
this.price = price;
}
@Override
public int compareTo(MoonCake o) {
double gap = o.price - this.price;
if(gap > 0)
{
return 1;
}else if(gap < 0){
return -1;
}else {
return 0;
}
}
}