拼多多校招题: 小熊吃糖
时间限制:1秒
空间限制:32768K
有n只小熊,他们有着各不相同的战斗力。每次他们吃糖时,会按照战斗力来排,战斗力高的小熊拥有优先选择权。前面的小熊吃饱了,后面的小熊才能吃。每只小熊有一个饥饿值,每次进食的时候,小熊们会选择最大的能填饱自己当前饥饿值的那颗糖来吃,可能吃完没饱会重复上述过程,但不会选择吃撑。
现在给出n只小熊的战斗力和饥饿值,并且给出m颗糖能填饱的饥饿值。
求所有小熊进食完之后,每只小熊剩余的饥饿值。
输入描述:
第一行两个正整数n和m,分别表示小熊数量和糖的数量。(n <= 10, m <= 100)
第二行m个正整数,每个表示着颗糖能填充的饥饿值。
接下来的n行,每行2个正整数,分别代表每只小熊的战斗力和当前饥饿值。
题目中所有输入的数值小于等于100。
输出描述:
输出n行,每行一个整数,代表每只小熊剩余的饥饿值。
输入例子1:
2 5
5 6 10 20 30
4 34
3 35
输出例子1:
4
0
例子说明1:
第一只小熊吃了第5颗糖
第二只小熊吃了第4颗糖
第二只小熊吃了第3颗糖
第二只小熊吃了第1颗糖
思路:对于m个糖,无疑是要进行排序的,这样每次熊才能迟到最大的糖,对于n个bear,用Map结构存储对应的战斗力,和食量,便于访问。 因为要按照战斗力大小分进食顺序,所以要对map进行排序改造,将map节点存储在list中,重写Collections类的sort方法。 同时在最后输出的时候是按照输入的顺序来输出每一个熊剩余饥饿值的,也就是按照插入顺序输出,(不是按照熊的战斗力来输出哦),所以我们的map要定义为LinkedHashMap,而不是HashMap(hashmap是无序的,遍历出的顺序并不是拟输入的顺序)。其次在逻辑上我们只需要按照每只熊的战斗力顺序访问存储糖的数组即可,在考虑好定义的结构后,理清逻辑后,就可以写代码了。
public static void test5(){
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
List<Integer>list = new ArrayList<>();
for(int i = 0;i < m;i++){
int candy = sc.nextInt();
list.add(candy);
}
Collections.sort(list, new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
int []array = new int[m];
for(int i = 0;i < m;i++){
array[i] = list.get(i);
}
LinkedHashMap<Integer,Integer>map = new LinkedHashMap<>();
for(int i = 0;i < n;i++){
int battle = sc.nextInt();
int hungry = sc.nextInt();
map.put(battle,hungry);
}
Map<Integer,Integer> save = map;
map = sortHashMap(map);
for(Map.Entry<Integer,Integer>entry : map.entrySet()){
int key = entry.getKey();
int value = entry.getValue();
for(int i = 0;i < m;i++){
int cur = array[i];
if(cur != 0 && value >= cur){
value -= cur;
array[i] = 0;
}
}
map.put(key,value);
}
for(Map.Entry<Integer,Integer>entry : save.entrySet()){
System.out.println(map.get(entry.getKey()));
}
}
public static LinkedHashMap<Integer,Integer> sortHashMap(LinkedHashMap<Integer,Integer> map){
Set<Map.Entry<Integer,Integer>> set = map.entrySet();
List<Map.Entry<Integer,Integer>>list = new ArrayList<>(set);
Collections.sort(list, new Comparator<Map.Entry<Integer, Integer>>() {
@Override
public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
return o2.getKey() - o1.getKey();
}
});
LinkedHashMap<Integer,Integer> linkedHashMap = new LinkedHashMap<>();
for(Map.Entry<Integer,Integer>entry : list){
linkedHashMap.put(entry.getKey(),entry.getValue());
}
return linkedHashMap;
}
public static void main(String[] args) {
test5();
}