题目描述
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;
}