首页 > 代码库 > zoj 3659 并查集
zoj 3659 并查集
http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4882
现在在牡丹江,明天regional现场赛,不出一个月就要退役了,求保佑
今天热身赛做题很紧张,老是打错字,所以晚上写写代码练练手
脑子还是不好使,没想到可以并查集
思路:题目中的操作导致的一个结果是,一条边会成为一个集合的w,---- 如果能想到这里可能就能想到并查集吧
WA了因为如果father[x]==x并不能表示x中只有一个数啊。。。
#include <cstdio> #include <cstring> #include <algorithm> #include <iostream> using namespace std; #define ll long long #define IN(s) freopen(s,"r",stdin) const int MAXN = 200000+200; struct edge{ int u,v,w; bool operator < (const edge &c)const{ return w>c.w; } }e[MAXN*2]; int n,fa[MAXN],num[MAXN]; ll co[MAXN]; int Find(int x) { if(x != fa[x])fa[x]=Find(fa[x]); return fa[x]; } void init() { for(int i=0;i<=n;i++) fa[i]=i,num[i]=1,co[i]=0LL; } void un(int x,int y,int w) { int xf=Find(x); int yf=Find(y); if(xf == yf)return; /*if(xf == x && yf == y && num[xf]==1 && num[yf]==1) { num[xf]+=num[yf]; fa[yf]=xf; co[xf]=(ll)w*num[yf]+co[xf]; return; }*/ ll cosx=(ll)w*num[yf]+co[xf]; ll cosy=(ll)w*num[xf]+co[yf]; if(cosx > cosy) { fa[yf]=xf; num[xf]+=num[yf]; co[xf]=cosx; //co[y]+=w*num[x]; return; } fa[xf]=yf; num[yf]+=num[xf]; co[yf]=cosy; } ll solve() { init(); for(int i=1;i<n;i++) { un(e[i].u,e[i].v,e[i].w); } return co[Find(1)]; } int main() { //IN("zoj3659.txt"); while(~scanf("%d",&n)) { for(int i=1;i<n;i++) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w); sort(e+1,e+n); printf("%I64d\n",solve()); } return 0; }
zoj 3659 并查集
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。