【BZOJ5499】[2019省队联测]春节十二响（贪心）
题面
题解
如果是一条折链，显然维护两侧的值，每次两个堆分别弹出一个(max)然后合并一下，最后再放回去就可以了。
那么现在回到一棵树上，可以认为就是本身有一条链，现在每次要合并一条链进来，那么拿一个堆维护这个合并过程就可以了。为了保证复杂度用启发式合并。
在(C++11)下可以直接使用(.swap())函数来进行优先队列的交换。
为了在(BZOJ)上过就写的普通的启发式合并。
#include<cstdio>
#include<vector>
#include<queue>
using namespace std;
#define MAX 200200
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
int n,M[MAX];long long ans;
priority_queue<int> S[MAX];
struct Line{int v,next;}e[MAX];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int ID[MAX];
void dfs(int u)
{
ID[u]=u;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;dfs(v);
if(S[ID[v]].size()>S[ID[u]].size())swap(ID[u],ID[v]);
vector<int> tmp;
while(!S[ID[v]].empty())tmp.push_back(max(S[ID[u]].top(),S[ID[v]].top())),S[ID[u]].pop(),S[ID[v]].pop();
while(!tmp.empty())S[ID[u]].push(tmp.back()),tmp.pop_back();
}
S[ID[u]].push(M[u]);
}
/*
void dfs(int u)
{
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;dfs(v);
if(S[v].size()>S[u].size())S[u].swap(S[v]);
vector<int> tmp;
while(!S[v].empty())tmp.push_back(max(S[u].top(),S[v].top())),S[u].pop(),S[v].pop();
while(!tmp.empty())S[u].push(tmp.back()),tmp.pop_back();
}
S[u].push(M[u]);
}
*/
int main()
{
n=read();
for(int i=1;i<=n;++i)M[i]=read();
for(int i=2;i<=n;++i)Add(read(),i);
dfs(1);
while(!S[ID[1]].empty())ans+=S[ID[1]].top(),S[ID[1]].pop();
printf("%lld
",ans);
return 0;
}