标签:ch int 十二 long 春节 isdigit include getchar
省选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\n",ans);
return 0;
}
标签:ch,int,十二,long,春节,isdigit,include,getchar 来源: https://www.cnblogs.com/karryW/p/10806885.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。