poj3253
2015-07-10 21:03
337 查看
**## 贪心法 ##**
开始的思路是:
1.N个数据排序
2.将最小的两个数据相加的和与剩下的数据合在一起组成N-1个数据,并将最小的两个数据相加的和存储在res中。
3.重复1,2直到剩下一个数据。
但是出现了Time Limit Exceeded
#include <iostream> #include <algorithm> #define MAX_N 20000 using namespace std; int L[MAX_N],N; long long ans=0; void slove() { int j,temp,mi=0; while(N-mi>1){ L[mi+1]+=L[mi]; ans+=L[mi+1]; sort(L+mi+1,L+N); mi++; } } int main() { cin>>N; for(int i=0;i<N;i++) cin>>L[i]; sort(L,L+N); slove(); cout<<ans<<endl; return 0; }
随后 进行了修改,将每次将两个最小的数之和插入以排好序的数组中。
#include <iostream> #include <algorithm> #define MAX_N 20000 using namespace std; int L[MAX_N],N; long long ans=0; void slove() { int j,temp,mi=0; while(N-mi>1){ L[mi+1]+=L[mi]; ans+=L[mi+1]; temp=L[mi+1]; for(int i=mi+2;i<N;i++){ //进行插入 if(temp<L[i]){ for(j=mi+1;j<i-1;j++) L[j]=L[j+1]; L[j]=temp; break; } if(i==N-1){ for(j=mi+1;j<i;j++) L[j]=L[j+1]; L[j]=temp; } } mi++; } } int main() { cin>>N; for(int i=0;i<N;i++) cin>>L[i]; sort(L,L+N); slove(); cout<<ans<<endl; return 0; }
此题还可以用最优队列解。
//poj 3253 (优先队列) #include <iostream> #include <queue> #define MAX_N 20000 using namespace std; int L[MAX_N],N; void slove() { long long res=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 t1=que.top(); que.pop(); int t2=que.top(); que.pop(); que.push(t1+t2); res+=t1+t2; } cout<<res<<endl; } int main() { cin>>N; for(int i=0;i<N;i++) cin>>L[i]; slove(); return 0; }
相关文章推荐