题目:
农夫约翰为了修理栅栏,要将一块很长的木板分割成N块。准备切成的木板的长度为L1、L2、……、Ln. 未切割木板的长度恰好为切割木板的长度和。每次切断木板时,需要的开销为这块木板的长度。例如,长度为21的木板切割成5、8、8的三块木板。长为21的木板切割成13、8时,开销为21.再将长度为13的木板切割成长度5、8时,开销为13.于是合计开销为34。于是按题目要求将木板切割完,最小的开销是多少?
限制条件:
1<=N<=2000
0<=Li<=5000
输入样例:
N=3, L={8, 5, 8}
输出样例:
34
思路:用朴素的方法实现的话,时间复杂度为O(n^n)。因为只需要从板的集合中取出最短的两块,并且把长度为两块长度之和的板加入集合中即可,所有使用优先队列就可以高效的实现。一共需要进行O(n)次O(log N)的操作,因此总的时间复杂度为O(Nlogn)。
优先队列法(构造小根堆):
#include<bits/stdc++.h>
#include<queue>
using namespace std;
#define MAX_N 2000
typedef long long ll;
int N;
int L[MAX_N+1];
void solve()
{
ll ans=0;
priority_queue<int ,vector<int>,greater<int> > que;//申明一个从小到大取出数值的优先队列
for(int i=0;i<N;i++){
que.push(L[i]);
}
while(que.size()>1)
{
int l1,l2;
l1=que.top();
que.pop();
l2=que.top();
que.pop();
ans+=l1+l2;
que.push(l1+l2);
}
printf("%lld\n",ans);
}
int main()
{
cin>>N;
for(int i=0;i<N;i++) cin>>L[i];
solve();
}
/*
3
8 5 8
*/