Source Code
#include<iostream>
#include<bits/stdc++.h>
typedef long long ll;
typedef long double ld;
const int mod=1e9+7;
const ll INF=1e18;
const double pi=3.14159265358979323846;
#define F first
#define S second
#define all(v) v.begin(),v.end()
#define pb push_back
#define Go ios_base::sync_with_stdio(false); cin.tie(nullptr); cout.tie(nullptr);

using namespace std;

int n;
ll v[200010],ans[200010],dif[200010];
int p[200010];

int root(int u,int k)
{
    if(k==4)return u;
    else return root(p[u],k+1);
}

int main()
{
Go

cin >> n;
for(int i=2;i<=n;i++){
    int x; cin >> x;
    p[i]=x;
}
for(int i=1;i<=n;i++)cin >> v[i];

for(int i=1;i<=n;i++)ans[i]=1e18,dif[i]=1e18;

for(int i=1;i<=n;i++){
    int pa=root(i,0);
    if(pa==0)continue;
    if(dif[pa]>abs(v[i]-v[pa])){
        dif[pa]=abs(v[i]-v[pa]);
        ans[pa]=i;
    }
    else if(dif[pa]==abs(v[i]-v[pa])&&ans[pa]>i)
        ans[pa]=i;
}

for(int i=1;i<=n;i++)if(ans[i]==1e18)ans[i]=0;

for(int i=1;i<=n;i++)
    cout << ans[i] << ' ';

return 0;
}
Copy
Find a Friend yumna08
GNU G++17
48 ms
7.8 MB
Accepted