【十二省联考2019】【贪心】春节十二响

【题目背景】

“清明时节雨纷纷，路上行人欲断魂。”

2075年的清明没有春雨。在漫天飞雪的笼罩下，穿行在冰原间的，只有载着人类微薄希望的雪地车。

遥遥 4.22 光年的征途，对于地球这孤独的旅人而言，恐怕也是无比寂寞吧。

【题目描述】

距离苏拉威西只有一百公里了，车内的空气比窗外更加冰冷。四双眼睛紧盯着艾莉 芬面前的屏幕，那是控制行星发动机的关键程序：春节十二响。他需要将其部署到电力控制系统的一个芯片中。

“春节十二响”由 n个子程序构成，第 i 个子程序所需的内存空间是 Mi。这 n 个子程序之间的调用关系构成了一棵以第 1 个子程序为根的树，其中第 i个子程序在调用树上的父亲是第 fi个子程序。

由于内存紧张，电力控制芯片上提供了一种内存分段机制。你可以将内存分为若干 个段 S1 , S2,…,Sk ，并将每个程序预先分配到一个固定的段。如果两个子程序没有直接或间接的调用关系，则他们可以被分配到同一个段中，反之则不能。换言之，当且仅当 a 和b 在调用树上不是祖先—后代关系，a 和 b 可以被分配到同一个段中。

一个段的大小应当是所有分配到这个段的子程序所需内存大小的最大值，所有段 大小的和不能超过系统的内存大小。

现在艾莉芬想要知道，电力控制芯片至少要有多少内存，才能保证春节十二响的正 确运行。即：最少需要多大的内存，才能通过先将内存分成若干个段，再把每个子程序分配到一个段中，使得每个段中分配的所有子程序之间不存在祖先—后代关系。

【输入】
第一行包含一个正整数 n 表示子程序的个数，其中n⩽2×105 。

第二行有n 个用空格隔开的正整数 M1 ,M2,…,Mn，Mi 表示第 i 个子程序所需的内存空间。

第三行有 n−1个用空格隔开的正整数f2,f3,…,fn，满足fi<i，表示第i 个子程序在调用树上的父亲是第fi 个子程序。

【输出】
仅一个整数，表示最小的内存需求。

【样例输入】
5
10 20 20 30 30
1 1 2 2
【样例输出】
60

【思路】

这道题很有意思。我们首先考虑一条链的情况。那么只有在根节点时，我们才需要对内存段进行合并。对于这两颗子树，我们把内存段从大到小排序，依次合并显然是最优的。同样的结论也适用于树上。我们枚举当前节点的每个儿子，把它们的内存从大到小依次合并起来，即：
$$no{w}_{k^{\prime },i}=max(no{w}_{k,i},no{w}_{son,i})$$
其中，k是已经考虑的儿子，son是还没有考虑的儿子，i就是内存段从大到小排序的下标，now是当前状态下的内存段情况。这样类似树形dp依次合并上去，到根就行了。但是还有一个问题，这样合并的复杂度是$O(n^2)$。注意到一次合并的复杂度和少的那一部分集合大小有关，而集合大小是与深度有关的。所以我们就可以借鉴长链剖分的思想，把所有信息合并到最深的那个儿子上，这样总共合并$O(n)$次，一次合并可以用优先队列O(logn)实现。所以总时间复杂度为$O(nlogn)$。这道题是一个好题，和NOIP2018赛道修建有异曲同工之妙，都是从特殊的情况（菊花树，链）开始考虑，最后推广到一般情况。
代码：

#include<bits/stdc++.h>
#define re register
using namespace std;
const int N=2e5+5;
int n,m,a,b,c;
inline int red(){
    int data=0;char ch=0;
    ch=getchar();
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') data=(data<<3)+(data<<1)+ch-'0',ch=getchar();
    return data;
}
int val[N],dep[N],son[N],id[N],nw[N],cnt=0;
priority_queue<int>q[N];
vector<int>g[N];
long long ans=0;
int main(){
	n=red();
	for(int re i=1;i<=n;++i)val[i]=red();
	for(int re i=2;i<=n;++i)g[red()].push_back(i);
	for(int re i=1;i<=n;++i)id[i]=i;
	for(int re i=n;i;--i){
		int pos=0;
		for(int re j=g[i].size()-1;~j;--j)
			if(dep[g[i][j]]+1>dep[i])pos=j,dep[i]=dep[g[i][j]]+1;
		if(g[i].size()>1)swap(g[i][0],g[i][pos]);//找出最深的儿子
		for(int re j=g[i].size()-1;j>0;--j){//其它儿子依次与最深的儿子合并
			cnt=0;
			while(!q[id[g[i][j]]].empty())
			nw[++cnt]=max(q[id[g[i][j]]].top(),q[id[g[i][0]]].top()),
			q[id[g[i][j]]].pop(),q[id[g[i][0]]].pop();
			for(int re j=1;j<=cnt;j++)q[id[g[i][0]]].push(nw[j]);
		}
		if(g[i].size())id[i]=id[g[i][0]];//记录最深的儿子的堆的编号，方便父亲调用
		q[id[i]].push(val[i]);
	}
	while(!q[id[1]].empty())ans+=q[id[1]].top(),q[id[1]].pop();
	cout<<ans<"\n";
}