CUMTOJ:algorithm-锯木棒（题解）

题目描述

xiaok大佬最近再雇佣工人给他掰木棒。把一根长为L的木棒锯成两段，他需要支付给工人L元钱。xiaok大佬一开始只有长为L的一根木棒，他想把它锯成n段，每段长度分别为L1,L2,...,Ln，问xiaok大佬最少要付给工人多少钱？

输入

第一行两个整数n,L(1<n<10的3次方,n<L<10的9次方) 
第二行n个整数L1,L2,...,Ln（0<Li<L,且保证L1+L2+...+Ln=L） 

输出

输出一个整数，表示最小花费

样例输入

3 21
8 5 8

样例输出

34

算法描述

本题可以逆序拼接木头，使用贪心算法。每次从现有的木棒中选取两个长度最短的木棒进行拼接，将拼接后的木棒插入到原来的木棒序列中，重新排好序，同时将每次的花费进行累加，当只剩一根木棒时，得到的累加和即为题目所求。

代码如下

#include <iostream>
#include <vector>
#include<algorithm>
using  namespace std;
int main()
{
    int n, m;
    cin >> n >> m;
    vector<int>a;//存放当前各段木棒长度
    for (int i = 0; i < n; i++)
    {
        int t; cin >> t;
        a.push_back(t);
    }
    sort(a.begin(), a.end());
    int* b = new int[n];
    int top = 0;
    while (a.size() > 1)
    {
        int temp = a[0] + a[1];
        b[top++] = temp;
        vector<int>::iterator k = a.begin();//每次选取两个长度最短的木棒拼接
        a.erase(k);
        k = a.begin();
        a.erase(k);
        int i;
        for ( i = 0; i < a.size(); i++)
            if (a[i] > temp)
                break;
        a.insert(a.begin() + i, temp);//将拼接后的木棒按序插入到存放木棒的容器中
    }
    int value = 0;
    for (int i = 0; i < top; i++)
        value += b[i];
    cout << value;
    return 0;
}