当前位置:   article > 正文

走廊泼水节(TYVJ1391)_走廊泼水节tyvj

走廊泼水节tyvj

题面:给定一颗N个节点的树,要求增加若干条边,把这棵树扩充为完全图,并满足图的唯一最小生成树仍然是这棵树。

           求增加的边的权值总和最小是多少。

数据范围:N<=6000,边权均为非负整数。 

基本思路:把N-1条边按权值从小到大排序,一次扫描每条树边,执行kruskal(在每一步加一个操作)。

设当前扫描到的边(x,y,v),若x,y不在一个集合中,则合并Sx,Sy,代价是(v+1)*(|Sx|*|Sy|-1)。这样就保证了每个集合都是一个完全图,(v+1)则保证了原树为图的最小生成树

如图   1234已生成集合S1,  567已生成集合S2。目前扫到的最小的原树上的边为(3,5,v),而合并S1,S2的代价就为(v+1)*(3*4-1)。连好之后的图就变为:

 

还是一个完全图耶。

so,本题是水水的kruskal,好好领悟bia。时间复杂度(n logn)。

  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. #define ll long long
  4. int n,T,fa[6010],s[6010];
  5. ll ans;
  6. struct rec{int x,y,v;}e[6010];
  7. bool mycmp(rec a,rec b) {return a.v<b.v;}
  8. int gf(int x)//找爹
  9. {
  10. if(x==fa[x]) return x;
  11. return fa[x]=gf(fa[x]);
  12. }
  13. int main()
  14. {
  15. // freopen("data.in","r",stdin);
  16. // freopen("data.out","w",stdout);
  17. ios::sync_with_stdio(false);
  18. cin>>n;
  19. ans=0;
  20. for(int i=1;i<n;i++)
  21. cin>>e[i].x>>e[i].y>>e[i].v;
  22. sort(e+1,e+n,mycmp);//按边权从小到大排序
  23. for(int i=1;i<=n;i++)
  24. fa[i]=i,s[i]=1;//并查集初始化
  25. for(int i=1;i<n;i++)
  26. {
  27. int x=gf(e[i].x);
  28. int y=gf(e[i].y);
  29. if(x==y) continue;
  30. //如果不在一个集合里,累加答案,合并。
  31. ans+=(ll)(e[i].v+1)*(s[x]*s[y]-1);
  32. fa[x]=y;
  33. s[y]+=s[x];
  34. }
  35. cout<<ans<<endl;
  36. return 0;
  37. }

 HLOJ 1523...(忽略它)

 

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/52828
推荐阅读
相关标签
  

闽ICP备14008679号