2023牛客寒假算法基础集训营2-D 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;
}