省选Day2T2
自己的思路:
将树上所有的链存到好几个数组里,每次取出每个数组的最大值比较,将最大的计入答案。其实很接近正解了。
.
正解:将每个点的子树合成一个堆,启发式合并(由小到大)。
上代码吧
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define LL long long
using namespace std;
const int N = 200000+10;
int n,a[N],fa[N],tmp[N],id[N],tim;
int head[N],tot;
struct edge{
int node,next;
}e[N];
priority_queue<int> q[N];
inline int read()
{
register int x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
return (f==1)?x:-x;
}
void add(int x,int y)
{
e[++tot].next=head[x];
head[x]=tot;
e[tot].node=y;
}
void dfs(int x)
{
id[x]=++tim;
for(int i=head[x];i;i=e[i].next)
{
int v=e[i].node;
dfs(v);
if(q[id[x]].size()<q[id[v]].size())
swap(id[x],id[v]);
int m=q[id[v]].size();
for(int i=1;i<=m;i++)
{
tmp[i]=max(q[id[x]].top(),q[id[v]].top());
q[id[x]].pop();
q[id[v]].pop();
}
for(int i=1;i<=m;i++)
q[id[x]].push(tmp[i]);
}
q[id[x]].push(a[x]);
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
a[i]=read();
for(int i=2;i<=n;i++)
fa[i]=read(),add(fa[i],i);
dfs(1);
LL ans=0;
while(!q[id[1]].empty())
ans+=q[id[1]].top(),q[id[1]].pop();
printf("%lld
",ans);
return 0;
}