题目链接:除2!
题目描述:
给一个数组,一共有 n个数。你能进行最多 k次操作。每次操作可以进行以下步骤:
选择数组中的一个偶数 ai,将其变成 ai/2。 现在你进行不超过 k次操作后,
让数组中所有数之和尽可能小。请输出这个最小的和。
算法原理:
由于题目说明要使的数组的和最小,并且不能超过k次,那我们要明白数组的和最小,我们每次要取出数组中最大的偶数来进行除2操作,并且除2之后如果这个数还是最大的,我们要一直循环操作;这样不难想到我们学过优先级队列,将偶数放入优先级队列中并按照大根堆进行排序,这样变能保证我们每次操作的偶数都是最大的,附上AC代码
题目代码:
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Scanner;
public class Main {
// 输入一组整数,偶数可以进行除2操作做,在进行不超过k次除2操作,让这些数的和尽可能最小
public static void main(String[] args) {
// 创建优先级队列并按大堆排列
PriorityQueue<Long> priorityQueue = new PriorityQueue<>(new Comparator<Long>() {
@Override
public int compare(Long o1, Long o2) {
return Math.toIntExact((o2 - o1));
}
});
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
long k = scanner.nextLong();
// 变量t来保留每次除2之后的数
long t, tmp, sum = 0;
// count变量来记录除2的次数
long count = 0;
// 将奇数进行相加,偶数放入优先级队列中
for (int i = 0; i < n; i++) {
tmp = scanner.nextLong();
if (tmp % 2 != 0) {
sum += tmp;
} else {
priorityQueue.offer(tmp);
}
}
// 如果队列为空,则全是奇数
if (priorityQueue.isEmpty()) {
System.out.println(sum);
} else {
while (true) {
count++;
// 当count大于k之后不符合需求,直接返回
if (count > k) {
break;
}
// 取出队首元素并删除,进行除2操作
t = priorityQueue.peek();
priorityQueue.poll();
t /= 2;
// 除2后再次进行判断是否为偶数,是的话加入队列
if (t % 2 != 0) {
sum += t;
} else {
priorityQueue.offer(t);
}
// 队列为空直接返回
if (priorityQueue.isEmpty()) {
break;
}
}
// k次操作后再次判断队列是否为空
while (!priorityQueue.isEmpty()) {
t = priorityQueue.poll();
sum += t;
}
System.out.println(sum);
}
}
}