贪心&合并果子

题意不用多说了。

思路:哈夫曼编码的思路,保证每次合并的两队果子分别为所有堆得最小和次小。注意直接一次排序递加的思路是很危险的0.0.

下面上代码:

#include
#include
#include
#include
#include
#include
#include
using namespace std;
typedef long long LL;
//typedef pair P;
//const int  maxn=100000;
int ans[1005];
LL sum=0;
void solve(int x)
{
    sum=0;
    while(x>1)
    {
        int minn1=0,minn2=1;
        if(ans[minn1]>ans[minn2]) swap(minn1,minn2);
        for(int i=2; i<x; i++)
        {
            if(ans[i]<ans[minn1])
            {
                minn2=minn1;
                minn1=i;
            }
            else if(ans[i]<ans[minn2]) minn2=i;
        }
        sum+=ans[minn1]+ans[minn2];
        if(minn1==x-1) swap(minn1,minn2);
        ans[minn1]=ans[minn1]+ans[minn2];
        ans[minn2]=ans[x-1];
        x--;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n;
        scanf("%d",&n);
        for(int i=0; i<n; i++)
            scanf("%d",&ans[i]);
        solve(n);
        printf("%lld\n",sum);

    }
    return 0;
}

你可能感兴趣的:(ACM&贪心大法)