[十二省联考2019]春节十二响

[十二省联考2019]春节十二响

好后悔考场没思考这道最良心的水题...t1花了我两个半点结果连20都拿不到

哭了

题目描述：略

思路：将一个点下面的几条链合并成一条

 

用长链剖分的话似乎可以简化操作

实在太水了所以没啥好说的...

#include<cstdio>
#include<queue>
using std::priority_queue;
typedef long long lint;
const int N=200011;
template<typename st> st max(st a,st b){return a>b?a:b;}
int n;
lint v[N];
struct sumireko
{
    int to,ne;
}e[N];
int he[N],ecnt;
void addline(int f,int t)
{
    e[++ecnt].to=t;
    e[ecnt].ne=he[f];
    he[f]=ecnt;
}
int fa[N],len[N],top[N],dson[N];
void dfs1(int x)
{
    len[x]=1;
    for(int i=he[x],t;i;i=e[i].ne)
    {
        t=e[i].to;
        if(t==fa[x]) continue;
        dfs1(t);
        if(len[t]>=len[x])
        {
            dson[x]=t;
            len[x]=len[t]+1;
        }
    }
}
priority_queue<lint> q[N];
lint sta[N],hop;
void dfs2(int x,int tt)
{
    top[x]=tt;
    if(dson[x])
    {
        dfs2(dson[x],tt);
        for(int i=he[x],t;i;i=e[i].ne)
        {
            t=e[i].to;
            if(t==fa[x]||t==dson[x]) continue;
            dfs2(t,t);
            for(int j=1;j<=len[t];j++)
            {
                lint a=q[tt].top(),b=q[t].top();
                q[tt].pop(),q[t].pop();
                sta[++hop]=max(a,b);
            }
            while(hop)
            {
                q[tt].push(sta[hop--]);
            }
        }
    }
    q[tt].push(v[x]);
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&v[i]);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&fa[i]);
        addline(fa[i],i);
    }
    dfs1(1);
    dfs2(1,1);
    lint ans=0;
    while(!q[1].empty())
    {
        ans+=q[1].top();
        q[1].pop();
    }
    printf("%lld\n",ans);
    return 0;
}