Codeforces 1059E. Split the Tree

题目：http://codeforces.com/problemset/problem/1059/E

用倍增可以在nlog内求出每个节点占用一个sequence 时最远可以向父节点延伸到的节点，对每个节点作为sequence 的最后一个元素向上延伸时，将节点的父节点属性合并（类似于并查集的操作），

存在优先队列里保证每次先操作dep最大的节点

#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<string.h>
#include<cstring>
#include<algorithm>
#include<set>
#include<map>
#include<fstream>
#include<cstdlib>
#include<ctime>
#include<list>
#include<climits>
#include<bitset>
#include<random>
#include <ctime>
#include <cassert>
#include <complex>
#include <cstring>
#include <chrono>
using namespace std;
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define fopen freopen("input.in", "r", stdin);freopen("output.in", "w", stdout);
#define left asfdasdasdfasdfsdfasfsdfasfdas1
#define set asfdasdasdfasdfsdfasfsdfasfdas2
#define tan asfdasdasdfasdfasfdfasfsdfasfdas
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef long long ll;
typedef unsigned int un;
const int desll[][]={{,},{,-},{,},{-,}};
const int mod=1e9+;
const int maxn=1e5+;
const int maxm=1e5;
const double eps=1e-;
const int csize=;
int n,k,m,ar[maxn];
int f[maxn],tan[maxn][],v[maxn],in[maxn],mark;
ll ss[maxn][];
bool ma[maxn],que[maxn];
priority_queue<pair<int,int> > qu;
vector<int> ve[maxn];
int dep[maxn];
void dfs(int u,int pre)
{
    if(pre>=)dep[u]=dep[pre]+;
    for(int i=;i<ve[u].size();i++){
        int v=ve[u][i];
        if(v==pre)continue;
        dfs(v,u);
    }
}
int main()
{
    int l;
    ll s,mx=;f[]=;
    scanf("%d%d%I64d",&n,&l,&s);
    memset(in,,sizeof(in));
    memset(tan,,sizeof(tan));
    memset(ma,,sizeof(ma));
    memset(que,,sizeof(que));
    for(int i=;i<=n;i++)scanf("%d",&ar[i]),mx=max(mx,1LL*ar[i]);
    for(int i=;i<=n;i++)scanf("%d",&f[i]),in[f[i]]++,ve[f[i]].push_back(i);
    dep[]=;
    dfs(,-);
    if(mx>s){
        printf("-1\n");
        return ;
    }
    for(int i=;i<=n;i++){
        tan[i][]=f[i];
        ss[i][]=ar[f[i]];
    }
    for(int j=;j<;j++){//倍增预处理
        for(int i=;i<=n;i++){
            if(i + (<<j) >n)break;
            tan[i][j]=tan[tan[i][j-]][j-];
            ss[i][j] = ss[i][j-]+ss[tan[i][j-]][j-];
        }
    }
    for(int i=;i<=n;i++){
        ll lmid=l-,smid=s-ar[i];
        int ins=,x=i;
        while(ins>=){
            if(tan[x][ins]==)ins--;
            else if((<<ins) > lmid)ins--;
            else if(ss[x][ins]<=smid){
                smid-=ss[x][ins];
                lmid -= (<<ins);
                x=tan[x][ins];
            }
            else ins--;
        }
        v[i]=x;//v[i]代表i节点作为sequence尾节点时最远可以向上延伸到的节点
    }
    for(int i=;i<=n;i++){
        if(v[i]==)exit();
    }
    for(int i=;i<=n;i++){
        if(in[i]==){
            qu.push(make_pair(dep[i],i));
            que[i]=;
        }
    }
    int ans=;
    while(qu.size()){
        int u=qu.top().second;qu.pop();
        if(ma[u])continue;//如果已经标记过，代表有其他节点可以延伸到此节点，跳过即可
        mark=v[u];
        int mid=f[u],pre=u;
        while(dep[mid]>dep[mark]){//对ma数组进行标记，合并f数组，类似于并查集的操作
            ma[pre]=;
            f[pre]=mark;
            pre=mid;
            mid=f[mid];
        }
        ma[pre]=;
        ma[mark]=;
        if(pre!=mark)f[pre]=mark;
        ans++;
        //cout<<mark<<" "<<f[mark]<<" "<<ma[f[mark]]<<" "<<que[f[mark]]<<endl;
        if(f[mark]> && !ma[f[mark]] && !que[f[mark]]){
            qu.push(make_pair(dep[f[mark]],f[mark]));
            que[f[mark]]=;
        }
    }
    printf("%d\n",ans);

    return ;
}