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;
}