BZOJ4756

最新推荐文章于 2022-09-16 19:24:51 发布

Gipsyu

最新推荐文章于 2022-09-16 19:24:51 发布

阅读量285

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/Gipsy_Danger/article/details/79701713

版权

数据结构专栏收录该内容

54 篇文章 0 订阅

订阅专栏

其实一开始我没想出来

只是想出了dfs序

但后边就卡壳了

然后看了题解恍然大悟

其实就是一道挺水的题

你先把dfs序找出来

用树状数组来统计

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

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

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

很有意思

#include<bits/stdc++.h>
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<ll> 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<<P[i].id;
//    }
//    cout<<endl;
//    for(int i=1;i<=2*n;++i){
//        cout<<A[i];
//    }
//    cout<<endl;
    for(ll i=1;i<=2*n;++i){
        if(B[A[i]]==0){
            C[A[i]]=-(sum(N)-sum(P[A[i]].id));
            B[A[i]]=1;
        }
        else{
            C[A[i]]+=sum(N)-sum(P[A[i]].id);
            Add(P[A[i]].id,1);
        }
    }
    for(ll i=1;i<=n;++i){
        cout<<C[i]<<endl;
    }
    return 0;
}
//34125
//1355324421