寒假训练营第五天(贪心)A题:合并果子

A - 合并果子

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

题解:直接贪心即可,先对每堆果子按数量排序,两个最小的相加。然后再进行排序,又对排序后的最小两堆果子相加…

AC代码如下:

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;

int main()
{
    int T,k,minSum,n;
    scanf("%d",&T);
    int a[105];
    while(T--)
    {
        minSum=0;
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }
        sort(a,a+n);
        k=a[0];
        for(int i=0;i<n-1;i++)
        {
            a[i]=k;
            sort(a+i,a+n-1);
            k=a[i];
            k+=a[i+1];
            minSum+=k;
        }
        printf("%d\n",minSum);
    }
    return 0;
}

小白心得

猜你喜欢

转载自blog.csdn.net/boliu147258/article/details/86644951