xtu1045唯一的最小生成树

Description

求一个非负权边的无向连通图的最小生成树，如果这个无向图存在两个或两个以上的最小生成树，就输出Not Unique，否则输出最小生成树的边的权值和。

输入：

第一行是一个整数K，表示有多少个测试用例，以后每个测试用例占m+1行。每个测试用例的第一行为两个整数n，m（3<=n<=100），表示图的顶点数和边数，从第二行开始每行为三个整数i，j,w,表示从i到j顶点的权值。

输出：

每行输出一个测试用例的结果。如果这个无向图存在两个或两个以上的最小生成树，就输出Not Unique，否则输出最小生成树的边的权值和。

Sample Input

2
3 3
1 2 1
2 3 2
3 1 3
4 4
1 2 2
2 3 2
3 4 2
4 1 2

Sample Output

3
Not Unique

Source
程序设计实践
确定最小生成树是不是唯一只要先把最小生成树找出来，然后去掉一条边后在剩下的边里寻找出最小生成树，如果结果与之前一样，那么最小生成树不唯一。

#include<cstdio>#include<cmath>#include<cstring>#include<iostream>#include<algorithm>#include<queue>#include<vector>#include<map>#include<stack>#define pi acos(-1.0)#define EPS 1e-6#define INF (1<<24)using namespace std;struct EDGE{    int x,y;    int dis;}edge[10005];int father[105];int n,m;bool flag[10005];bool cmp(struct EDGE a,struct EDGE b){    return a.dis<b.dis;}int findfather(int x){    if(x!=father[x])        father[x]=findfather(father[x]);    return father[x];}void Uion(int x,int y){    int a=findfather(x);    int b=findfather(y);    father[a]=b;}bool same(int x,int y){    if(findfather(x)==findfather(y)) return true;    else return false;}int kruskal(){    int i;    for(i=0;i<=n;i++) father[i]=i;    int res=0;    for(i=0;i<m;i++)    {        if(same(edge[i].x,edge[i].y)==false)        {            Uion(edge[i].x,edge[i].y);            res+=edge[i].dis;            flag[i]=true;  //标记是否为最小生成数中的边        }    }    return res;}int kruskal2(int p){    int i;    for(i=0;i<=n;i++) father[i]=i;    int res=0;    for(i=0;i<m;i++)    {        if(i==p) continue;  //如果是去掉的那条边直接略过。        if(same(edge[i].x,edge[i].y)==false)        {            Uion(edge[i].x,edge[i].y);            res+=edge[i].dis;        }    }    return res;}int main(){    int k;    scanf("%d",&k);    while(k--)    {        int i,j;        scanf("%d %d",&n,&m);        for(i=0;i<m;i++) scanf("%d %d %d",&edge[i].x,&edge[i].y,&edge[i].dis);        sort(edge,edge+m,cmp);        memset(flag,false,sizeof(flag));        int out=kruskal();        int flagt=true;        for(int i=0;i<m;i++){            if(flag[i]==true){                //edge[i].dis=INF;                if(kruskal2(i)==out){                    flagt=false;                    break;                }            }        }        if(flagt==true) printf("%d\n",out);        else printf("Not Unique\n");    }    return 0;}