1. 程式人生 > >合併果子(貪心+優先佇列)

合併果子(貪心+優先佇列)

現在有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

分析:
貪心的思想
每次選擇價值最小和次小的那兩個進行合併,合併成新的果子放進果堆中,然後又在果堆中選擇最小的和次小的進行合併

這樣下來,合併所有果子所花費的代價肯定是最小的
所以我們可以用優先佇列來實現這個過程
代價最小的果子優先

code:
#include<stdio.h>
#include <iostream>
#include <math.h>
#include <queue>
using namespace std;
int main()
{
    int t,n,a;
    scanf("%d",&t);
    while(t--)
    {
        priority_queue<int,vector<int>,greater<int
> >q;//優先佇列,數值小的優先 scanf("%d",&n); for(int i=0;i<n;i++) { scanf("%d",&a); q.push(a); } int ans=0,temp; while(q.size()>=2)//果堆中只有一個果子的時候合併就完成了 { temp=0; temp+=q.top(),q.pop(); temp
+=q.top(),q.pop();//每次去優先佇列隊首的兩個,因為他們是代價最小和次小的 ans+=temp; q.push(temp);//將合併後的果子再次放入堆中 } printf("%d\n",ans); } return 0; }