Hdu 5782 Cycle(改变一个人的能力值(变为100000),使得整个公司的能力值的和最大)

传送门:Hdu 5782 Cycle

题意:给你一个公司的架构,为一棵树,每个人的能力值是它子树的中位数(包括这个点),你现在可以改变一个人的能力值(变为100000),使得整个公司的能力值的和最大

思路:如果一个节点的能力值改变,那么它的子树中有一个能力值小于等于现在的当前节点的能力值改变
那么我们只要从根到叶子进行dfs(先求出每个节点的能力值和其中位数变化之后增大的值),
如果访问到了这个节点,那么将这个点的能力值作为位置,中位数变化量作为累加量,插入到树状数组中
对于每个点维护一个树状数组的后缀和的最大值

#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
vector<int>G[maxn];
int siz[maxn],tot,st[maxn],ed[maxn];
int Mid1[maxn],Mid2[maxn];
const int MAXN=maxn*30;
struct node{
    int l,r,w;
}T[MAXN];
int sz,maxv;
int a[maxn],root[maxn];
long long C[maxn];

void update(int &i,int l,int r,int num){
    T[++sz]=T[i],i=sz;
    T[i].w++;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    if(num<=mid)
        update(T[i].l,l,mid,num);
    else
        update(T[i].r,mid+1,r,num);
}

int query(int x,int y,int l,int r,int k){
    if(l==r)
        return l;
    int mid=(l+r)>>1;
    if(T[T[y].l].w-T[T[x].l].w>=k)
        return query(T[x].l,T[y].l,l,mid,k);
    else
        return query(T[x].r,T[y].r,mid+1,r,k-(T[T[y].l].w-T[T[x].l].w));
}

void dfs(int u){
    siz[u]=1,st[u]=++tot;
    root[tot]=root[tot-1];
    update(root[tot],1,maxv,a[u]);
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        dfs(v);
        siz[u]+=siz[v];
    }
    ed[u]=tot;
}

long long Max;

void add(int x,int num){
    while(x>0)  C[x]+=num,x-=(x&-x);
}

long long sum(int x){
    long long ans=0;
    while(x<=maxv)  ans+=C[x],x+=(x&-x);
    return ans;
}

void dfs1(int u){
    Max=max(Mid2[u]-Mid1[u]+sum(a[u]),Max);
    add(Mid1[u],Mid2[u]-Mid1[u]);
    for(int i=0;i<G[u].size();i++){
        int v=G[u][i];
        dfs1(v);
    }
    add(Mid1[u],-Mid2[u]+Mid1[u]);
}

int main(){
    int n,u;
    while(scanf("%d",&n)!=EOF){
        maxv=0,tot=0,sz=0,root[0]=0;
        memset(C,0,sizeof(C));
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]),G[i].clear(),maxv=max(maxv,a[i]);
        if(n==1){
            printf("100000\n");
            continue;
        }
        for(int i=2;i<=n;i++)
            scanf("%d",&u),G[u].push_back(i);
        dfs(1);
        long long ans=0;
        for(int i=1;i<=n;i++){
            if(siz[i]==1){
                Mid1[i]=a[i];
                Mid2[i]=100000,ans+=Mid1[i];
                continue;
            }
            int k=(siz[i]+1)/2;
            Mid1[i]=query(root[st[i]-1],root[ed[i]],1,maxv,k);
            Mid2[i]=query(root[st[i]-1],root[ed[i]],1,maxv,k+1);
            ans+=Mid1[i];
        }
        Max=0;
        dfs1(1);
        printf("%lld\n",Max+ans);
    }
    return 0;
}