【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[N],rch[N],size[N];
long long sum[N];
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;
}