BZOJ4756

其实一开始我没想出来

只是想出了dfs序

但后边就卡壳了

然后看了题解恍然大悟

其实就是一道挺水的题

你先把dfs序找出来

用树状数组来统计

第一次入栈的时候统计比他大的树

第二次出栈的时候统计比他大的树

然后第二次减去第一次就是答案

很有意思

#include
using namespace std;
typedef long long ll;
const ll N=2e5;ll c[N],A[N],B[N],C[N];
ll head[N],Next[N],ver[N];
ll tot=0;
void dfs(ll p){
    A[++tot]=p;
    for(ll i=head[p];i;i=Next[i]){
        dfs(ver[i]);
    }
    A[++tot]=p;
}
void add(ll x,ll y){
    ver[++tot]=y;
    Next[tot]=head[x];
    head[x]=tot;
}
ll lowbit(ll i){return i&(-i);}
void Add(ll i,ll value){while(i<=N){c[i]+=value;i+=lowbit(i);}}
ll sum(ll i){ll sum=0;while(i>0){sum+=c[i];i-=lowbit(i);}return sum;}
vector V;
ll getid(ll x){//离散化
    return lower_bound(V.begin(),V.end(),x)-V.begin()+1;
}
struct room{
    ll p,id;
}P[N];
int main(){
    ll n,a;
    cin>>n;
    for(ll i=1;i<=n;++i){
        cin>>P[i].p;
        V.push_back(P[i].p);
    }
    sort(V.begin(),V.end());
    V.erase(unique(V.begin(),V.end()),V.end());
    for(ll i=1;i<=n;++i){
        P[i].id=getid(P[i].p);
    }
    for(ll i=2;i<=n;++i){
        cin>>a;
        add(a,i);
    }
    tot=0;
    dfs(1);
//    for(int i=1;i<=n;++i){
//        cout<