uva----10954

贪心，始终把数组里面最小的元素相加，想用stl里面数据结构做的，可惜今天过年，书丢在家了，晚上回去在研究一下

代码：

#include<iostream>
using namespace std;

int num[5000]={0};

int main()
{
    int N;
    while(cin>>N&&N)
    {
        for(int i=0;i<N;i++)
            cin>>num[i];

        int ans=0;
        while(N>1)
        {
            int a=0,b=1;
            if(num[a]>num[b])
            {
                int temp=a;
                a=b;
                b=temp;
            }
            for(int i=0;i<N;i++)
            {
                if(num[a]>num[i])
                    a=i;//NO.1min number
            }
            for(int i=0;i<N;i++)
            {
                if(num[b]>num[i]&&num[i]>=num[a]&&i!=a)
                    b=i;
            }

            ans+=num[a]+num[b];

            if(a!=(N-1)&&b!=(N-1))
            {
                num[a]=num[a]+num[b];
                num[b]=num[N-1];
            }
            else if(a==(N-1))
                num[b]=num[a]+num[b];
            else if(b==(N-1))
                num[a]=num[a]+num[b];

            N--;
        }
        cout<<ans<<endl;
    }
    return 0;
}

 

STL（优先队列）做法：

#include<iostream>
#include<queue>
using namespace std;

int main()
{
    int N,num;
    while(cin>>N&&N)
    {
        int ans=0;
        priority_queue<int, vector<int>, greater<int> >que;//声明队列，从小到大
        for(int i=0;i<N;i++)
        {
            cin>>num;
            que.push(num);
        }

        while(que.size()>1)
        {
            int l,l_;
            l=que.top();
            que.pop();
            l_=que.top();
            que.pop();

            ans+=(l+l_);
            que.push(l+l_);
        }

        cout<<ans<<endl;
    }
    return 0;
}

 

ps：

如果直接使用priority,省略三个参数中的两个，则默认是大的优先。

如果要用到小顶堆，则一般要把模板的三个参数都带进去。
STL里面定义了一个仿函数 greater<>，对于基本类型可以用这个仿函数声明小顶堆

priority_queue<int, vector<int>, greater<int> >que;