题意:
给了你n个点权,n-1条边(题目保证这个图是棵树),问你将这n-1条边都砍掉的花费和最小是多少。
砍掉一条边的花费是这条边所连的两个联通块最大的点权之和。
Solution:
因为要使最后砍完n-1条边之后的花费最小,可以贪心的知道,每次都是先将点权最大的点所连的边先砍掉,这样能够保证点权大的点被用到的次数尽可能的少。因此每次都先砍掉点权最大的点所连的边,然后对这条边的另一个联通块(即与点权最大的点所连的边的另外一个联通块)求一下点权最大,每次都重复这样的操作(删除的边不用再重复删除)。但是这样写的话,可能比较复杂,因此我们反着进行思考,将点权从小到大排序,依次加入该点,并遍历与该点所连的所有边,如果这条边已经加入,那么跳过,否则加入与该点相连,并且另外一个点的点权比该点小的边。
代码:
#include<bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f3f3f3f3f
typedef long long ll;
typedef pair<int,int>P;
int n;
int a[100005];
int f[100005];
int Find(int x){
if(f[x]!=x)
{
int tem=f[x];
f[x]=Find(f[x]);
a[f[x]]=max(a[tem],a[f[x]]);
a[tem]=a[f[x]];
}
return f[x];
}
void Union(int x,int y){
int u=Find(x),v=Find(y);
f[u]=v;
a[v]=max(a[u],a[v]);
a[u]=a[v];
}
vector<int> e[100005];
vector<P>g;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);g.push_back(P(a[i],i));}
sort(g.begin(),g.end());
for(int i=1;i<=n;i++)f[i]=i;
for(int i=0;i<n-1;i++)
{
int u,v;scanf("%d%d",&u,&v);
if(a[u]>a[v])swap(u,v);
e[v].push_back(u);
}
ll res=0;
for(int i=0;i<g.size();i++)
{
int u=g[i].second;
for(int j=0;j<e[u].size();j++)
{
int v=e[u][j];
if(Find(u)!=Find(v))
{
res+=a[Find(u)]+a[Find(v)];
Union(u,v);
}
}
}
printf("%lld",res);
return 0;
}