jzoj1082-合并果子【堆,贪心】

最新推荐文章于 2024-06-20 18:50:31 发布

转载最新推荐文章于 2024-06-20 18:50:31 发布 · 57 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/sslwyc/p/9218602.html

本文介绍了一道经典的贪心算法题目：如何将不同大小的果子堆通过最小的消耗合并为一堆。采用堆的数据结构来实现每次选取并合并最小的两堆，最终输出最小的总消耗。

题目

有n堆大小不同的果子堆，每次合并两堆消耗它们两堆的重量之和，求把它们全部合并的最小消耗

输入

输入包括两行，第一行是一个整数n(1<＝n<=10000)，表示果子的种类数。第二行包含n个整数，用空格分隔，第i个整数ai(1<＝ai<=20000)是第i种果子的数目。

输出

输出包括一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于2^31。

样例输入

3
1 2 9

样例输出

解题思路

贪心就是每次都合并最小的两个，这里用堆可以方便一些

代码

#include<cstdio>
using namespace std;
int a[10001],num,x,n;
long long s,u;
void up(int x)
{
    int t;
    while (x>1 && a[x]<a[x/2])
    {
        t=a[x];a[x]=a[x/2];a[x/2]=t;
        x/=2;
    }
}//调整堆
void down(int x)
{
    int t,y;
    while (x*2<=num && a[x]>a[x*2] || x*2+1<=num && a[x]>a[x*2+1])
    {
        y=x*2;
        if (x*2+1<=num && a[x*2]>a[x*2+1]) y++;
        t=a[x];a[x]=a[y];a[y]=t;
        x=y;
    }
}//调整堆
void insert(int x)
{a[++num]=x;up(num);}//插入元素
int main()
{
    scanf("%d",&n); 
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        insert(x);
    }//建堆
    while (num>1)
    {
        u=a[1];a[1]=a[num];num--;down(1);
        u+=a[1];a[1]=a[num];num--;down(1);//取两个最小的
        s+=u;num++;a[num]=u;up(num);//合并后放回
    }
    printf("%d",s);
}