正题
这题其实很简单。
很明显的一个贪心策略就是,优先处理深度较大的节点的答案。
因为深度更小的节点会有更多的决策可以决定,所以倒不如把优的决策让给自己的子孙。
然后很多人就直接线段树合并/可并堆来做到一个【常数较大的】log。
却没发现,这个可以直接把物品的价值排一遍序,然后将物品从大到小插入,每次处理离自己最近而且是祖先的决策。
那么这个东西可以直接用并查集维护。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
using namespace std;
int n,m;
int f[3000010],dad[3000010];
struct edge{
int y,next;
}s[3000010];
int first[3000010],len=0;
int op[3000010];
int findpa(int x){
if(f[x]==-1) return f[x];
if(f[x]!=x) return f[x]=findpa(f[x]);
return x;
}
void ins(int x,int y){
s[++len]=(edge){y,first[x]};first[x]=len;
}
void dfs(int x,int last){
if(op[x]) last=x;
f[x]=last;for(int i=first[x];i!=0;i=s[i].next) dfs(s[i].y,last);
}
int main(){
scanf("%d %d",&n,&m);
int x;
dad[0]=-1;
for(int i=1;i<=n-1;i++) scanf("%d",&x),ins(x,i),dad[i]=x;
for(int i=1;i<=n-1;i++) scanf("%d",&x),op[x]++;
dfs(0,-1);
long long ans=0;
for(int i=n-1;i>=0;i--){
int fx=findpa(i);
if(fx==-1) continue;
ans+=i;op[fx]--;
if(op[fx]==0) f[fx]=dad[fx];
}
printf("%lld\n",ans);
}