题目来源:[NWPU][2014][SRM][3]搜索与动规 A 题
http://vjudge.net/contest/view.action?cid=49918#problem/A
作者:npufz
题目:有一个很长的棒子要锯成指定数目和指定长度的段数,每锯一次的花费等于锯开的棒子的长度,由于锯断的方案很多,请找出花费的最小值
输入包括要锯成棒子的段数N,和每段的长度Li,棒子的总长度为各段的总和,1<=N<=20,000,1<=Li<=50,000
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
using namespace std;
int main()
{ int n,a[20003],i,m1,m2;
long long int sum;
cin>>n;
sum=0;
priority_queue< int ,vector<int>,greater<int> > que;
while(!que.empty()) que.pop();
for(i=0;i<n;i++)
{ scanf("%d",&a[i]);
que.push(a[i]);
}
while(que.size()>1)
{ m1=que.top();que.pop();
m2=que.top();que.pop();
sum+=(m1+m2);
que.push(m1+m2);
}
cout<<sum<<endl;
return 0;
}
反思:这个问题等价于把N段连接起来,每次连接两段,花费为所连接的两段的长度的和,这正是锯断的一个逆过程,可以在剩下的要连接的棒子里每次连接最短的两段,连接 N-1次就解决了,一开始用暴力搜索的方式每次找出两个最短的棒子,由于对步骤不够优化,就TLE了,因为N*N的复杂度是处在一个危险的边缘,稍微处理不好就容易
TLE,最后不想再优化了,就用了优先队列,复杂度降低了,就解决了