CSU 1588 合并果子 (贪心)

                                          合并果子

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

1.优先队列维护:

#include 
#include 
#include
#include
#include
#include
using namespace std;
int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        int n, v;
        priority_queue, greater >pq;
        scanf("%d", &n);
        for(int i = 0; i < n; i++){
            scanf("%d", &v);
            pq.push(v);
        }
        int res = 0;
        while(!pq.empty()){
            int v1 = pq.top();
            pq.pop();
            int v2 = pq.top();
            pq.pop();
            res += v1 + v2; //抓取最小的两堆合并
            if(!pq.empty()) pq.push(v1 + v2);
        }
        printf("%d\n", res);
    }
	return 0;
}

2.每次加和后生成新的果堆,对新的果堆重新排序选取最小的两堆加和

#include
#include
using namespace std;
int str[1005];
int main()
{
    int T,n,ans,k;
    cin>>T;
    while(T--)
    {
        ans=0;
        cin>>n;
        for(int i=0;i>str[i];
        }
        sort(str,str+n);
        k=str[0];
        for(int i=0;i<=n-2;i++)
        {
            str[i]=k;           //形成新果堆
            sort(str+i,str+n);  //新果堆重新排序
            k=str[i];           
            k+=str[i+1];        //新果堆最小两堆和相加
            ans+=k;             //统计结果
        }
        cout<

 

你可能感兴趣的:(贪心)