http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4882
原来一个排序加并查集就可以呀 思维还是不够强呀
这里没有环,所以并查集并不是用了处理环 而是指出当前集合中最优选择点
当两个集合合并时 将两个集合的最优选择点 比较择优更新
代码及其注释:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<vector>
#include<map>
#include<cmath>
#include<queue>
#include<stack>
#include<algorithm>
#define LL long long
using namespace std;
const int INF=0x3f3f3f3f;
const int N=200005;
struct node
{
int l,r;
LL d;
}mem[N];
LL ans[N];//如果为最优选择点 目前的值
LL num[N];//如果为最优选择点 目前集合的元素个数
int f[N];
bool cmp(node x,node y)
{
return x.d>y.d;
}
int findx(int x)
{
if(f[x]!=x)
f[x]=findx(f[x]);
return f[x];
}
int main()
{
//freopen("data.txt","r",stdin);
int n;
while(cin>>n)
{
for(int i=1;i<n;++i)
cin>>mem[i].l>>mem[i].r>>mem[i].d;
for(int i=1;i<=n;++i)
{f[i]=i;num[i]=1;}
sort(mem+1,mem+n,cmp);
memset(ans,0,sizeof(ans));
LL MAX=0;//最优答案
for(int i=1;i<n;++i)
{
int k1=findx(mem[i].l),k2=findx(mem[i].r);
ans[k1]+=(num[k2]*mem[i].d);
ans[k2]+=(num[k1]*mem[i].d);
//然后择优更新
if(ans[k1]>MAX) MAX=ans[k1];
if(ans[k2]>MAX) MAX=ans[k2];
if(ans[k1]>ans[k2])
{
num[k1]+=num[k2];
f[k2]=k1;
}else
{
num[k2]+=num[k1];
f[k1]=k2;
}
}
cout<<MAX<<endl;
}
return 0;
}
浙公网安备 33010602011771号