【BZOJ5499】[2019省队联测]春节十二响（贪心）

【BZOJ5499】[2019省队联测]春节十二响（贪心）

题面

BZOJ
洛谷

题解

若是是一条折链，显然维护两侧的值，每次两个堆分别弹出一个\(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\n",ans);
    return 0;
}