2809: [Apio2012]dispatching|splay启发式合并

ws_yzy

于 2016-03-06 15:59:45 发布

阅读量466

点赞数

分类专栏：平衡树文章标签： splay启发式合并

本文链接：https://blog.csdn.net/ws_yzy/article/details/50813790

版权

平衡树专栏收录该内容

2 篇文章 0 订阅

订阅专栏

倒着枚举每个忍者，找出他所在的集合中在总薪水<script type="math/tex" id="MathJax-Element-147"><=M</script>的情况下最多能过雇佣的人数，让当前的忍者做管理者然后更新答案，再合并他与上级所在的集合。
发现splay的启发式合并的内存是 $O(nlog_{2}n)$ 的，复杂度是 $O(nlog_{2}^2n)$

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
#include<set>
#include<map>
#include<iostream>
#include<algorithm>
#define ll long long
#define N 1000005
using namespace std;
int sc()
{
    int i=0,f=1; char c=getchar();
    while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9')i=i*10+c-'0',c=getchar();
    return i*f;
}
ll sum[N],l[N],mx,ans;
int root[N],fa[N],ch[N][2],w[N],v[N],b[N],size[N];
int n,cnt;
void push_up(int x)
{
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
    sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+v[w[x]];
}
void rotate(int x,int &f)
{
    int y=fa[x],z=fa[y],l,r;
    l=(ch[y][1]==x);r=l^1;
    if(z)ch[z][ch[z][1]==y]=x;
    ch[y][l]=ch[x][r],ch[x][r]=y;
    fa[ch[y][l]]=y;fa[y]=x;fa[x]=z;
    push_up(y),push_up(x);
    if(y==f)f=x;
}   
void splay(int x,int &f)
{
    while(x!=f)
    {
        int y=fa[x],z=fa[y];
        if(y!=f)
            if(ch[y][0]==x^ch[z][0]==y)rotate(x,f);else rotate(y,f);
        rotate(x,f);
    }
}
void insert(int &x,int f,int p,int k)
{
    if(!x)
    {
        w[x=++cnt]=p;
        fa[x]=f;
        size[x]=1;
        sum[x]=v[p];
        splay(x,root[k]);
        return ;
    }
    insert(ch[x][v[p]>v[w[x]]],x,p,k);
}
int find(int x,ll mx)
{
    if(!x)return 0;
    if(mx<sum[ch[x][0]])return find(ch[x][0],mx);
    int ans=size[ch[x][0]];
    mx-=sum[ch[x][0]];
    if(mx<v[w[x]])return ans;
    ans++,mx-=v[w[x]];
    return ans+find(ch[x][1],mx);
}
void dfs(int x,int f)
{
    if(!x)return;
    insert(root[f],0,w[x],f);
    dfs(ch[x][0],f);
    dfs(ch[x][1],f);
    //insert(root[f],0,w[x],f);
}
void merge(int x,int y)
{
    if(size[root[x]]<size[root[y]])swap(root[x],root[y]);
    dfs(root[y],x);
}
int main()
{
    n=sc(),mx=sc();
    for(int i=1;i<=n;i++)
        b[i]=sc(),v[i]=sc(),l[i]=sc(),insert(root[i],0,i,i);
    for(int i=n;i;i--)
    {
        int x=find(root[i],mx);
        ans=max(ans,(ll)x*l[i]);
        if(i!=1)merge(b[i],i);
    }
    cout<<ans;
    return 0;
}