2023牛客寒假算法基础集训营2-D Tokitsukaze and Energy Tree

2023牛客寒假算法基础集训营2-D Tokitsukaze and Energy Tree

Tokitsukaze and Energy Tree

样例输入

5
1 1 3 3
1 1 2 2 3

样例输出

22

---

题解：

本题涉及两个知识点：DFS+树的结论
该题是收获以X为根的子树的所有能量，因此整体先从整个树的下方往上方放(如果从上往下放就很浪费)
如果从1号点放置一个能量值为X的球，那么整个操作过程中共加了三次X的值。

即ans=∑depth_i * v_i（树i的深度×能量球i的值的所有总和）
这个公式与点积公式十分相似。
点积（内积）的定义：

而按照结论：两个向量内部允许任意重排，则点积（内积）最大方法为都升序排列。
因此对树的深度和能量球的值进行升序排序。

其实本题不需要以上线代的知识，贪心就行，短深度×小能量值，长深度×大能量值，这样就能保证和最大了。

#include<bits/stdc++.h>
#define pi 3.14159
#define ll long long 
#define ull unsigned long long
#define pii pair<int,int>
const int N=1e6+99;
#define MAX 0x3f3f3f3f
#define MAXX 0x3f3f3f3f3f3f3f3f
using namespace std;
#define db(x) cout<<#x<<'='<<x<<' ';
#define dbn(x) cout<<#x<<'='<<x<<endl;
#define dbf(x,p,q) for(ll i=p;i<=q;i++) { cout<<#x<<"["<<i<<"]"<<"="<<x<<' '};
ll n;
ll dep[N]={0},v[N]={0},fa[N]={0};
vector<ll> g[N];
void dfs(ll now,ll d)
{
	dep[now]=d;
	for(auto next:g[now])
	{
		dfs(next,d+1);
	}//计算树的深度
}
int main()
{
	cin>>n;
	for(int i=2;i<=n;i++)//从2开始，因为i=1时为根节点
	{
		scanf("%lld",&fa[i]);//根节点为fa[i]
		g[fa[i]].push_back(i);//对fa[i]进行操作
	}
	for(int i=1;i<=n;i++)
		scanf("%lld",&v[i]);
	dfs(1,1);
	sort(v+1,v+1+n);
	sort(dep+1,dep+1+n);
	ll ans=0;
	for(int i=1;i<=n;i++)
		ans+=dep[i]*v[i];
	cout<<ans<<endl;
	return 0;
}