解析
做法很多的一道题
sol1
先求出dfs序,离线下来,然后按权值大小的顺序统计答案并插到对应的dfs序中
sol2
离散化后,dfs过程中动态维护树状数组,利用前后差值求出答案
sol3
树上dsu
就比较无脑了,暴力维护即可
但是代码比前两个难写
sol4
权值线段树合并
这个做法在本题中确实是一点优势也没有
兼具了常数大和难编写的双重优势(bushi
总的来说,前两种做法比较推荐
代码
使用的是soltion1
别问我为什么当时还敲了个树剖和线段树上去
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=3e5+100;
int n,m;
struct node{int to,nxt;
}p[N<<1];
int fi[N],cnt=-1;
void addline(int x,int y){p[++cnt]=(node){y,fi[x]};fi[x]=cnt;
}
int dfn[N],fa[N],top[N],pos[N],tim,dep[N],hson[N],siz[N];
void dfs1(int x,int f){fa[x]=f;//printf("x=%d f=%d\n",x,f);siz[x]=1;for(int i=fi[x];~i;i=p[i].nxt){int to=p[i].to;if(to==f) continue;dep[to]=dep[x]+1;dfs1(to,x);siz[x]+=siz[to];if(!hson[x]||siz[hson[x]]<siz[to]) hson[x]=to;}return;
}
void dfs2(int x,int tp){dfn[++tim]=x;pos[x]=tim;top[x]=tp;if(hson[x]) dfs2(hson[x],tp);for(int i=fi[x];~i;i=p[i].nxt){int to=p[i].to;//printf("x=%d to=%d\n",x,to);if(to==fa[x]||to==hson[x]) continue;dfs2(to,to);}return;
}
#define mid ((l+r)>>1)
#define ls k<<1
#define rs k<<1|1
ll sum[N<<2],add[N<<2];
void pushup(int k){sum[k]=sum[ls]+sum[rs];
}
void Add(int k,int l,int r,ll v){add[k]+=v;sum[k]+=1ll*(r-l+1)*v;return;
}
void pushdown(int k,int l,int r){ll o=add[k];add[k]=0;if(!o) return;Add(ls,l,mid,o);Add(rs,mid+1,r,o);return;
}
void change(int k,int l,int r,int x,int v){if(l==r){sum[k]+=v;return;}pushdown(k,l,r);if(x<=mid) change(ls,l,mid,x,v);else change(rs,mid+1,r,x,v);pushup(k);return;
}
void longchange(int k,int l,int r,int x,int y,int v){if(x<=l&&r<=y){Add(k,l,r,v);return;}pushdown(k,l,r);if(x<=mid) longchange(ls,l,mid,x,y,v);if(y>mid) longchange(rs,mid+1,r,x,y,v);pushup(k);return;
}
ll ask(int k,int l,int r,int x,int y){if(x<=l&&r<=y){//printf("k=%d l=%d r=%d x=%d y=%d res=%lld\n",k,l,r,x,y,sum[k]);return sum[k];}pushdown(k,l,r);ll res=0;if(x<=mid) res+=ask(ls,l,mid,x,y);if(y>mid) res+=ask(rs,mid+1,r,x,y);//printf("k=%d l=%d r=%d x=%d y=%d res=%lld\n",k,l,r,x,y,res);return res;
}
ll sumquery(int x,int y){ll res=0;while(top[x]!=top[y]){if(dep[top[x]]<dep[top[y]]) swap(x,y);res+=ask(1,1,n,pos[top[x]],pos[x]);x=fa[top[x]];}if(dep[x]<dep[y]) swap(x,y);res+=ask(1,1,n,pos[y],pos[x]);return res;
}
struct P{int id,v;bool operator < (const P y)const{return v>y.v;}
}a[N];
int ans[N];
int main(){memset(fi,-1,sizeof(fi));scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%d",&a[i].v);a[i].id=i;}int x,y;for(int i=2;i<=n;i++){scanf("%d",&x);addline(x,i);addline(i,x);}dfs1(1,0);dfs2(1,1);//build(1,1,n);//for(int i=1;i<=n;i++) printf("i=%d dep=%d fa=%d siz=%d top=%d pos=%d hson=%d\n",i,dep[i],fa[i],siz[i],top[i],pos[i],hson[i]);//scanf("%d",&m);//int op;sort(a+1,a+1+n);for(int i=1;i<=n;i++){int now=a[i].id;ans[now]=ask(1,1,n,pos[now],pos[now]+siz[now]-1);change(1,1,n,pos[now],1);}for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
}
/**/