合併果子(優先佇列 +或者+哈夫曼)

尋找&星空の孩子發表於2015-05-01
佇列

合併果子

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 312  Solved: 113
[Submit][Status][Web Board]

Description

現在有n堆果子，第i堆有ai個果子。現在要把這些果子合併成一堆，每次合併的代價是兩堆果子的總果子數。求合併所有果子的最小代價。

 

Input

第一行包含一個整數T（T<=50），表示資料組數。
每組資料第一行包含一個整數n（2<=n<=1000），表示果子的堆數。
第二行包含n個正整數ai（ai<=100），表示每堆果子的果子數。

 

Output

每組資料僅一行，表示最小合併代價。

 

Sample Input

2
4
1 2 3 4
5
3 5 2 1 4

Sample Output

19
33

HINT

 

(優先佇列 +或者+哈夫曼)

 

#include<stdio.h>
#include<queue>
using namespace std;
struct node
{
   int a;
   friend bool operator<(node aa,node bb)
   {
       return aa.a>bb.a;
   }
};
int main()
{
    int n,t,a,sum;
    node bb,aa;
    scanf("%d",&t);
    while(t--)
    {
        priority_queue<node>q;
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a);
            bb.a=a;
            q.push(bb);
        }
        sum=0;
        while(!q.empty())
        {
            aa=q.top(); q.pop();//printf("%d ",aa.a);
            if(n==1)  break;
            bb=q.top(); q.pop();
            n--;
            aa.a+=bb.a;
            sum+=aa.a;
            q.push(aa);
        }
        printf("%d\n",sum);
    }
}
 
/**************************************************************
    Problem: 1588
    User: aking2015
    Language: C++
    Result: Accepted
    Time:92 ms
    Memory:1064 kb
****************************************************************/

 

相關文章