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;
}