本来以为这道题是01背包算法题,但是题干中有一句话:如果剩余时间不够做完一份卷子,可根据剩余时间获得卷子的部分价值。那么这道题其实就变成贪心算法踢了,只要计算出每张卷子的价值/时间,然后按这个降序排列,就能求出在给定的时间内的最大价值了。
package huawei;
import java.util.*;
/*
输入: int nPapers表示试卷的数目(1≤Papers≤20)
int nRemain表示剩余的时间(1≤nRemain≤10000),
int[][] paper是一个Papers*2的数组,每一行的两个元素依次为做完这一份试卷所需的时间、做完这份试卷的价值
List<Double> pMaxValue保证传入一个空的List,要求把获得的最大价值放入第一个位置
返回:
0:异常
1:计算成功返回
*/
public class Demo {
public int getMaxValue(int nPapers, int nRemain, int[][] paper,
List<Double> pMaxValue) {
int costTime = 0;
double maxValue = 0.00;
Double[] valuePerTime = new Double[nPapers];
Map<Integer, Double> map = new TreeMap<>();
List<Map.Entry<Integer, Double>> list = null;
if(nPapers <1 || nPapers > 20 || nRemain <1 || nRemain >10000) {
return 0;
}
if(paper.length != nPapers) {
return 0;
}
for(int i=0; i<nPapers; i++) {
if(paper[i][0] == 0 || paper[i][1] == 0) {
return 0;
}
int time = paper[i][0];
double value = paper[i][1];
valuePerTime[i] = value/time;
map.put(i, valuePerTime[i]);
}
list = new ArrayList<Map.Entry<Integer, Double>>(map.entrySet());
Collections.sort(list, new Comparator<Map.Entry<Integer, Double>>() {
@Override
public int compare(Map.Entry<Integer, Double> o1, Map.Entry<Integer, Double> o2) {
return o2.getValue().compareTo(o1.getValue());
}
});
for(Map.Entry<Integer, Double> a : list) {
if(costTime + paper[a.getKey()][0] < nRemain) {
costTime += paper[a.getKey()][0];
maxValue += paper[a.getKey()][1];
}else {
maxValue += (nRemain - costTime)*(a.getValue());
break;
}
}
pMaxValue.add(maxValue);
return 1;
}
}
做完提交,accept了,用例全部通过,但200分只得了198分,一脸懵逼中。。。。。。。