BZOJ 2809 [Apio2012]dispatching 可并堆

题意:链接

方法:可并堆

解析:

水题，但注意过程爆int.

方法就是找到树的根节点，之后扫，将每个子树什么的看做一个堆，然后之间合并，如果堆中的sum和超过了m，则去掉最大的，继续添加，这个显然啊。

然后每次处理完一个点，用堆中解更新答案。

比前两道水

代码:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100100
using namespace std;
typedef long long ll;
int n,m,all_root,cnt;
ll ans;
int head[N],h[N],fa[N],ch[N][2],root[N];
ll sum[N],val[N],lead[N],size[N];
struct node
{
    int from,to,next;
}edge[N];
int find(int x){return fa[x]==x?fa[x]:fa[x]=find(fa[x]);}
void pushup(int x){sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+val[x];size[x]=size[ch[x][0]]+size[ch[x][1]]+1;}
void init()
{
    memset(head,-1,sizeof(head)),cnt=1;
    for(int i=1;i<=n;i++)fa[i]=i;
}
void edgeadd(int from,int to)
{
    edge[cnt].to=to,edge[cnt].next=head[from],head[from]=cnt++;
}
int merge(int x,int y)
{
    if(!x)return y;
    if(!y)return x;
    if(val[x]<val[y])swap(x,y);
    ch[x][1]=merge(ch[x][1],y);
    if(h[ch[x][1]]>h[ch[x][0]])swap(ch[x][1],ch[x][0]);
    h[x]=h[ch[x][1]]+1;
    return x;
}
void dfs(int x)
{
    sum[x]=val[x],size[x]=1;
    for(int i=head[x];i!=-1;i=edge[i].next)
    {
        int to=edge[i].to;
        dfs(to);
        sum[x]+=sum[to];size[x]+=size[to];
        fa[x]=merge(fa[x],fa[to]);
    }
    while(sum[x]>m)
    {
        sum[x]-=val[fa[x]],size[x]--;
        fa[x]=merge(ch[fa[x]][0],ch[fa[x]][1]);
    }
    ans=max(lead[x]*size[x],ans);
}
int main()
{
    h[0]=-1;
    scanf("%d%d",&n,&m);
    init();
    for(int i=1;i<=n;i++)
    {
        int f,v,l;
        scanf("%d%d%d",&f,&v,&l);
        if(!f){all_root=i,val[i]=v,lead[i]=l,size[i]=1;continue;}
        edgeadd(f,i);
        val[i]=v,lead[i]=l,size[i]=1;
    }
    dfs(all_root);
    printf("%lld\n",ans);
}

版权声明：本文为博主原创文章，未经博主允许不得转载。