牛客AC题:除2!

题目链接:除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);
        }
    }
}

  • 10
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值