题目链接:https://codeforces.com/problemsets/acmsguru/problem/99999/507
题意:n个点,组成一个树,根为1。这个树有m个叶子节点,每个叶子节点上都有一个值。对于n-m个非叶子节点,输出其子树上的叶子节点中,每对节点值的差的绝对值的最小值。
思路:有mp维护子树中叶节点的值,每插入一个值时,用第一个比他小的值和第一个比他大的值和他作差,更新答案。
代码:
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 5e4+10;
const ll inf = (1LL<<31)-1;
int n,m,val[N];
ll ans[N];
struct node{
int to,nex;
}g[N];
int head[N],cnt;
void add(int u,int v){
g[++cnt].to=v;
g[cnt].nex=head[u];
head[u]=cnt;
}
int sz[N],son[N],SON,limit,siz;
map<int,int> mp;
map<int,int>::iterator it;
void getsz(int u){
sz[u]=1;
int mx=-1;
for(int i=head[u];i;i=g[i].nex){
getsz(g[i].to);
if(sz[g[i].to]>mx){
mx=sz[g[i].to];
son[u]=g[i].to;
}
sz[u]+=sz[g[i].to];
}
}
ll mn=inf;
void dfs1(int u){
if(u>=limit){
mp[val[u]]++;
if(mp[val[u]]==1){
it=mp.find(val[u]);
if(it!=mp.begin()){
it--;
mn=min(mn,0LL+val[u]-(*it).first);
it++;
}
it++;
if(it!=mp.end()) mn=min(mn,0LL+(*it).first-val[u]);
}else mn=0;
}
for(int i=head[u];i;i=g[i].nex)
if(g[i].to!=SON)
dfs1(g[i].to);
}
void dfs(int u,bool keep){
for(int i=head[u];i;i=g[i].nex){
if(g[i].to!=son[u]){
dfs(g[i].to,0);
}
}
if(son[u]){
dfs(son[u],1);
SON=son[u];
}
dfs1(u);
SON=0;
if(u<limit) ans[u]=mn;
if(!keep){
mp.clear();
mn=inf;
}
}
int main(void){
scanf("%d%d",&n,&m);
limit=n-m+1;
ans[1]=inf;
for(int i=2;i<=n;i++){
int u;
scanf("%d",&u);
add(u,i);
ans[i]=inf;
}
for(int i=limit;i<=n;i++)
scanf("%d",&val[i]);
getsz(1);
dfs(1,0);
printf("%lld",ans[1]);
for(int i=2;i<limit;i++)
printf(" %lld",ans[i]);
return 0;
}