提交时间:2024-03-31 09:24:37
运行 ID: 27821
#include<bits/stdc++.h> using namespace std; const int N = 1e6+10; int n,f[N],v[N],ans[N],vis[N],dep[N]; vector<int> G[N]; void dfs(int x,int fa){ dep[x]=dep[fa]+1; for(int i=0;i<G[x].size();i++){ int y=G[x][i]; if(y==fa) continue; dfs(y,x); } } void dfs2(int x,int fa,int bz){ if(dep[x]>=dep[bz]) ans[x]++; for(int i=0;i<G[x].size();i++){ int y=G[x][i]; if(y==fa) continue; if(vis[y]) continue; if(v[y]>bz){ vis[y]=1; return; } dfs2(y,x,bz); } } int main(){ bool flag=true; cin>>n; for(int i=2;i<=n;i++){ cin>>f[i]; G[f[i]].push_back(i); G[i].push_back(f[i]); if(f[i]!=(i-1)){ flag=false; } } for(int i=1;i<=n;i++){ cin>>v[i]; } if(!flag){ dfs(1,0); for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++) vis[j]=0; dfs2(1,0,i); } } else{ int maxn=0,bz=0; for(int i=1;i<=n;i++){ if(maxn<v[i]){ maxn=v[i]; bz=i; ans[i]++; } } } for(int i=1;i<=n;i++){ cout<<ans[i]<<" "; } cout<<endl; return 0; }