用优先队列解决围栏修复问题

一 问题描述

约翰想修牧场周围的篱笆，需要 N 块木板，每块木板都具有整数长度米。他购买了一块足够长的木板（长度为 Li 的总和，i 为 1 到 N ），以便得到 N 块木板。切割时木屑的损失可以不计。

农夫唐向约翰收取切割费用。切割一块木板的费用与其长度相同。切割长度为 21 米的木板需要 21 美分。唐让约翰决定切割长度的顺序和位置。约翰知道以不同的顺序切割木板，将会产生不同的费用。帮助约翰确定他得到 N 块木板的最低金额。

二 输入和输出

1 输入

第 1 行包含一个整数 N，表示木板的数量。第 2 到 N+1 行，每行包含一个所需木板的长度 Li。

2 输出

一个整数，即进行 N-1 次切割的最低花费。

三 输入和输出样例

1 输入样例

3

8

5

8

2 输出样例

34

四  算法设计

使用优先队列（最小值优先），每次都弹出两个最小值 t1、t2，t=t1+t2，sum+=t，将 t 入队，继续，直到队中只剩一个元素。sum 为所需花费。

算法实现：定义一个优先队列，输入元素入队。若队中只有一个元素，则直接累加输出即可。若队中多于一个元素，则每次都除出两个最小值，累加和值，并将和值入队。

五 代码

package com.platform.modules.alg.alglib.poj3253;

import java.util.PriorityQueue;

public class Poj3253 {
    public String output = "";

    public String cal(String input) {
        long sum;
        int n, t, t1, t2;

        PriorityQueue<Integer> q = new PriorityQueue<>(); // 默认为小顶堆
        String[] line = input.split("\n");
        n = Integer.parseInt(line[0]);
        for (int i = 1; i <= n; i++) {
            t = Integer.parseInt(line[i]);
            q.add(t);
        }
        sum = 0;
        if (q.size() == 1) {
            t1 = q.poll();
            sum += t1;
        }
        while (q.size() > 1) {
            t1 = q.poll();
            t2 = q.poll();
            t = t1 + t2;
            sum += t;
            q.add(t);
        }
        output=""+sum;
        return output;
    }
}

六 测试