【bzoj2809】[Apio2012]dispatching 主席树+dfs序

这分明就是一道主席树傻逼题呀，在dfs序上建出主席树，每次在主席树上二分就可以了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 101000
#define inf 1000000000
#define N 3400000

using namespace std;

int head[maxn],to[maxn],next[maxn],in[maxn],out[maxn],e[maxn];
int a[maxn],b[maxn];
int root[maxn],lch
,rch
,size
;
long long sum
;
int n,m,k,num,tot;

void dfs(int x)
{
e[++tot]=x;in[x]=tot;
for (int p=head[x];p;p=next[p]) dfs(to[p]);
out[x]=tot;
}

int modify(int pre,int l,int r,int x)
{
int now=++tot;
if (l==r)
{
lch[now]=rch[now]=0;size[now]=size[pre]+1;sum[now]=(long long)l+sum[pre];
}
else
{
int mid=(l+r)/2;
if (x<=mid)
{
lch[now]=modify(lch[pre],l,mid,x);rch[now]=rch[pre];
}
else
{
rch[now]=modify(rch[pre],mid+1,r,x);lch[now]=lch[pre];
}
size[now]=size[lch[now]]+size[rch[now]];
sum[now]=sum[lch[now]]+sum[rch[now]];
}
return now;
}

long long query(int root1,int root2,int l,int r,int m)
{
if (m<l) return 0;
if (sum[root2]-sum[root1]<=m) return size[root2]-size[root1];
if (l==r) return min(size[root2]-size[root1],m/l);
int mid=(l+r)/2;
if (sum[lch[root2]]-sum[lch[root1]]<=m) return size[lch[root2]]-size[lch[root1]]+query(rch[root1],rch[root2],mid+1,r,m-(sum[lch[root2]]-sum[lch[root1]]));
else return query(lch[root1],lch[root2],l,mid,m);
}

void addedge(int x,int y)
{
num++;to[num]=y;next[num]=head[x];head[x]=num;
}

int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
{
int x;
scanf("%d%d%d",&x,&a[i],&b[i]);
addedge(x,i);
}
dfs(1);
tot=0;
for (int i=1;i<=n;i++) root[i]=modify(root[i-1],1,inf,a[e[i]]);
long long ans=0;
for (int i=1;i<=n;i++) ans=max(ans,(long long)b[i]*query(root[in[i]-1],root[out[i]],1,inf,m));
printf("%lld\n",ans);
return 0;
}