poj3253

来源:互联网 发布:网络语言打麻将 编辑:程序博客网 时间:2024/06/05 14:44
              **## 贪心法 ##**

开始的思路是:
1.N个数据排序
2.将最小的两个数据相加的和与剩下的数据合在一起组成N-1个数据,并将最小的两个数据相加的和存储在res中。
3.重复1,2直到剩下一个数据。
但是出现了Time Limit Exceeded

#include <iostream>#include <algorithm>#define MAX_N 20000using 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 20000using 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 20000using 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;}
0 0
原创粉丝点击