codeforce1059 E. Split the Tree 树状数组+2遍DFS

codeforce514 E. Split the Tree

划分一棵以1为根的树，每个划分中不超过L个节点，边的权重不超过S

在DFS1的同时树状数组以深度维护该节点到根节点的权值之和，每次二分查找路径上他能到达最远的一个节点，

DFS2找出最小划分

#include<bits/stdc++.h>
using namespace std;
const int MAX=1e5+5;
int n,L,ans,len[MAX];
long long a[MAX],c[MAX],b[MAX],S;
struct P
{
    int to,nxt;
}e[MAX];
int head[MAX],tot;
void init()
{
    memset(head,-1,sizeof(head));
    memset(c,0,sizeof(c));
    memset(b,0,sizeof(b));
    tot=0;
}
inline void adde(int u,int v)
{
    e[tot].to=v;e[tot].nxt=head[u];head[u]=tot++;
}
inline void add(int x,long long v)
{
    while(x<=n) c[x]+=v,x+=x&(-x);
}
inline long long get(int r)
{
    long long sum=0;
    while(r) sum+=c[r],r-=r&(-r);
    return sum;
}
inline long long get(int l, int r)
{
    return get(r)-get(l-1);
}
int find_(int dep)
{
    int l=1,r=dep,mid;
    while(l<r)
    {
        mid=(l+r)>>1;
        long long sum=get(mid,dep);
        if(sum>S) l=mid+1;
        else if(sum<S) r=mid;
        else {r=mid;break;}
    }
    return min(L,dep-r+1);
}
void dfs1(int u,int dep)
{
    add(dep,a[u]-b[dep]);b[dep]=a[u];
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        dfs1(v,dep+1);
    }
    len[u]=find_(dep);
}
int dfs2(int u)
{
    int left=0;
    for(int i=head[u];~i;i=e[i].nxt)
    {
        int v=e[i].to;
        left=max(left,dfs2(v));
    }
    if(left==0) {left=len[u];++ans;}
    return left-1;
}
int main()
{
    int x;
    while(~scanf("%d%d%lld",&n,&L,&S))
    {
        long long max_=0;
        init();ans=0;
        for(int i=1;i<=n;++i) {scanf("%lld",&a[i]);max_=max(max_,a[i]);}
        for(int i=2;i<=n;++i) {scanf("%d",&x);adde(x,i);}
        if(max_>S) {printf("-1\n");continue;}
        if(L==1) {printf("%d\n",n);continue;}
        dfs1(1,1);dfs2(1);
        printf("%d\n",ans);
    }
    return 0;
}