数据结构与算法(Java) 44:最小分割代价

题目 一块金条切成两半,是需要花费和长度数值一样的铜板的。比如长度为20的 金条,不管切成长度多大的两半,都要花费20个铜板。一群人想整分整块金条,怎么分最省铜板? 例如,给定数组{10,20,30},代表一共三个人,整块金条长度为 10+20+30=60. 金条要分成10,20,30三个部分。 如果, 先把长度60的金条分成10和50,花费60再把长度50的金条分成20和30,花费50 一共花费110铜板。 但是如果, 先把长度60的金条分成30和30,花费60再把长度30金条分成10和20,花费30一共花费90铜板。 输入一个数组,返回分割的最小代价。

思路 建立小根堆。每次从堆中取出最小的两个值,计算加和,并将其添加到代价中,再将加和加入到小根堆中,直至小根堆中只剩下一个元素停止。【切割时就从上往下切】(贪心)

package algorithm.section7;

import java.util.Comparator;
import java.util.PriorityQueue;

public class Less_Money {
    public static int lessMoney(int[] arr){
        if (arr.length == 0) return 0;
        PriorityQueue<Integer> queue = new PriorityQueue<>();
        int sum = 0;

        for (int value : arr) queue.add(value);

        while (!queue.isEmpty()){
            int min1 = queue.poll();
            if (!queue.isEmpty()) {
                int min2 = queue.poll();

                sum += min1;
                sum += min2;

                queue.add(min1 + min2);
            }
        }

        return sum;
    }

    public static class minHeapComparator implements Comparator<Integer>{
        @Override
        public int compare(Integer o, Integer t1) {
            return o - t1;
        }
    }

    public static class maxHeapComparator implements Comparator<Integer>{

        @Override
        public int compare(Integer integer, Integer t1) {
            return t1 - integer;
        }
    }

    public static void main(String[] args){
        int[] arr = {6, 7, 8, 9};
        System.out.println(lessMoney(arr));

        int[] arr1 = {3, 5, 2, 7, 0, 1, 6, 4};
        PriorityQueue<Integer> queue1 = new PriorityQueue<>(new minHeapComparator());
        for (int value : arr1) queue1.add(value);
        while (!queue1.isEmpty()) System.out.print(queue1.poll() + " ");
        System.out.println();

        PriorityQueue<Integer> queue2 = new PriorityQueue<>(new maxHeapComparator());
        for (int value : arr1) queue2.add(value);
        while (!queue2.isEmpty()) System.out.print(queue2.poll() + " ");
    }
}

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值