bzoj2809: [Apio2012]dispatching（DFS序+主席树）

题目传送门
。

解法：
对于每个节点。
选他做管理员。
满意度为领导力*子树最多选多少人。
需解决子树最多选多少人。

考虑主席树。
既然是子树。肯定要用到DFS序。
DFS之后子树的编号都是连续的。
那么我们相当于在区间内求数最多且和不超过M。

按照权值建线段树（也就是主席树）
如果左子树的总费用是大于等于当前预算的话。
那么肯定去左子树搜索。
因为左子树是权值较小的（主席树以权值来建线段树）
如果当前左子树总费用小于当前预算的话。
那么肯定去右子树找答案+左子树总人数。

代码实现：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
typedef long long ll;
struct trnode {int lc,rc;ll s,c;trnode(){s=c=0;}}t[2100000];int cnt,n,rt[110000];
ll A[110000],B[110000],C[110000];int end[110000];
bool cmp(int a,int b) {return A[a]<A[b];}
void build(int &u,int l,int r,int p,ll s) {
    if(u==0)u=++cnt;
    t[u].c++;t[u].s+=s;int mid=(l+r)/2;
    if(l==r)return ;
    if(p<=mid)build(t[u].lc,l,mid,p,s);
    else build(t[u].rc,mid+1,r,p,s);
}
void Merge(int &u1,int u2) {
    if(u1==0){u1=u2;return ;}
    if(u2==0)return ;
    t[u1].c+=t[u2].c;t[u1].s+=t[u2].s;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
ll find(int u1,int u2,int l,int r,ll k,ll s) {
    if(l==r) {
        if(t[u1].s-t[u2].s<=k&&t[u1].c-t[u2].c>0)return s+1;
        else return s;
    }
    ll c=t[t[u1].lc].s-t[t[u2].lc].s;int mid=(l+r)/2;
    if(c<=k) return find(t[u1].rc,t[u2].rc,mid+1,r,k-c,s+t[t[u1].lc].c-t[t[u2].lc].c);
    else return find(t[u1].lc,t[u2].lc,l,mid,k,s);
}
struct node {int x,y,next;}a[110000];int len,last[110000];
void ins(int x,int y) {len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;}
int z,st[110000],ed[110000],ys[110000];
void dfs(int x) {
    st[x]=++z;ys[z]=x;
    for(int k=last[x];k;k=a[k].next)dfs(a[k].y);
    ed[x]=z;
}
int sy[110000];
int main() {
    ll m;scanf("%d%lld",&n,&m);cnt=0;
    len=0;memset(last,0,sizeof(last));
    for(int i=1;i<=n;i++) {
        int x;scanf("%d%lld%lld",&x,&A[i],&B[i]);
        if(x!=0)ins(x,i);C[i]=A[i];end[i]=i;
    }
    z=0;dfs(1);sort(end+1,end+1+n,cmp);
    for(int i=1;i<=n;i++) sy[end[i]]=i;
    for(int i=1;i<=n;i++) {
        build(rt[i],1,n,sy[ys[i]],A[ys[i]]);
        Merge(rt[i],rt[i-1]);
    }
    ll ans=0;
    for(int i=1;i<=n;i++) {
        ll sum=find(rt[ed[i]],rt[st[i]-1],1,n,m,0);
        ans=max(ans,sum*B[i]);
    }printf("%lld\n",ans);
    return 0;
}