合并果子(贪心算法)

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

贪心思路:

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

#include<iostream>
#include<algorithm>
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<n;i++)
        {
            cin>>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<<ans<<endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq1013459920/article/details/81317514
今日推荐