【洛谷P1090】合并果子

合并果子

题目链接

 

贪心：每次先合并最小的两堆果子

用堆实现

 

#include<iostream>
#include<cstdio>
using namespace std;
int heap_size,n;
int heap[30001];

void swap(int &a,int &b)
{
    int t=a;a=b;b=t;
}

void put(int d)
{
    int now,next;
    heap[++heap_size]=d;
    now=heap_size;
    while(now>1)
    {
        next=now>>1;
        if(heap[now]>=heap[next]) return;
        swap(heap[now],heap[next]);
        now=next;
    }
}

int get()
{
    int now,next,res;
    res=heap[1];
    heap[1]=heap[heap_size--];
    now=1;
    while(now<<1<=heap_size)
    {
        next=now<<1;
        if(next<heap_size&&heap[next+1]<heap[next]) next++;
        if(heap[now]<heap[next]) return res;
        swap(heap[now],heap[next]);
        now=next;
    }
    return res;
}

void work()
{
    int i,x,y,ans=0;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        cin>>x;
        put(x);
    }
    for(i=1;i<n;i++)
    {
        x=get();
        y=get();
        ans+=x+y;
        put(x+y);
    }
    cout<<ans<<endl;
}

int main()
{
    ios::sync_with_stdio(false);
    work();
    return 0;
}

手写堆真恶心。。

STL是个好东西

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

priority_queue<int,vector<int>,greater<int> >h;

int n,x,ans=0,tmp=0;

int main() {

    scanf("%d",&n);
    for(int i = 1; i <= n; i++) {
        scanf("%d",&x);  h.push(x);                           
    }    
    for(int i = 1; i <= n-1; i++) {
        tmp = h.top( );                     
        h.pop( );
        tmp = tmp+h.top( );                  
        h.pop( );
        h.push(tmp);                          
        ans += tmp;                        
    }
    printf("%d",ans);       
    return 0;
}