标签:min 方法 break name 相互 get 思路 ++ sample
天气晴朗的魔法
这样阴沉的天气持续下去,我们不免担心起他的健康。
Input两个正整数N,M。(1 <= N <= 10^5, N <= M <= 2 * 10^5)
接下来M行,每一行有三个整数A, B, V。(1 <= A, B <= N, INT_MIN <= V <= INT_MAX)
保证输入数据合法。Output输出一个正整数R,表示符合条件的魔法阵的魔力值之和。Sample Input
4 6 1 2 3 1 3 1 1 4 7 2 3 4 2 4 5 3 4 6
Sample Output
12
思路:开始先排序,想用二分来枚举最大边求最大生成树,思路应该没问题,不知为啥总WA。。后来又想到了一种方法,先求出最小生成树,get第n-1条边,然后再生成一棵所有边都不大于get边的最大生成树。保证最大变最小,且能相互连通。(极端情况为两棵树为同一棵树)
#include<stdio.h> #include<algorithm> #define MAX 100005 #define INF 100000000000000000 using namespace std; typedef long long ll; int f[MAX]; struct Node{ int u,v,w; }edge[2*MAX]; bool cmp1(Node a,Node b) { return a.w<b.w; } bool cmp2(Node a,Node b) { return a.w>b.w; } int find(int x) { return f[x]==x?x:f[x]=find(f[x]); } ll kru1(int n,int m) { int i; for(i=1;i<=n;i++){ f[i]=i; } sort(edge+1,edge+m+1,cmp1); int cnt=0;ll x=0; for(i=1;i<=m;i++){ int u=edge[i].u; int v=edge[i].v; int w=edge[i].w; int fu=find(u),fv=find(v); if(fu!=fv){ f[fv]=fu; cnt++; if(cnt==n-1) x=w; } if(cnt==n-1) break; } if(cnt<n-1) return -1; else return x; } ll kru2(int n,int m,int max) { int i; for(i=1;i<=n;i++){ f[i]=i; } sort(edge+1,edge+m+1,cmp2); int cnt=0;ll x=0; for(i=1;i<=m;i++){ int u=edge[i].u; int v=edge[i].v; int w=edge[i].w; if(w<=max){ int fu=find(u),fv=find(v); if(fu!=fv){ x+=w; f[fv]=fu; cnt++; } } if(cnt==n-1) break; } if(cnt<n-1) return -1; else return x; } int main() { int n,m,u,v,w,i; scanf("%d%d",&n,&m); for(i=1;i<=m;i++){ scanf("%d%d%d",&u,&v,&w); edge[i].u=u; edge[i].v=v; edge[i].w=w; } ll maxx=0,ans=0; maxx=kru1(n,m); ans=kru2(n,m,maxx); printf("%lld\n",ans); return 0; }
标签:min 方法 break name 相互 get 思路 ++ sample
原文地址:http://www.cnblogs.com/yzm10/p/7296384.html