poj 1679 Kruskal+MST环性质

题意：在一个无向图中，若其最小生成树唯一，则输出其值；若不唯一则输出“Not Unique!”。

《算法：C语言实现》这本书将MST有两个重要性质：割性质和环性质。本题则用到了MST的环性质。也就是说在一个无向连通图中可以将边分为MST上的边和非MST上的边，若将非MST上的边e加入MST中一定会形成一个环。所以先用Kruskal算法求出MST，然后以e的一个端点开始在MST上进行深搜，找到到达另一个端点路径，查看这条路径上边的权值是否有和e的权值相等的。若有则不是唯一的，若将每个e都测试一遍都没有和e的权值相等的则是唯一的。这样说够通俗易懂了吧……然而我看到网上很多人用次小生成树的解法来做，这种解法当然是正确的。但本题只是让判断唯一不唯一，本人认为没必要那样做，我的思路已经说过了，以下是我的代码：

#include<iostream>

#include<cstring>

#include<vector>

using namespace std;

struct node

{

	int start;

	int end;

	int value;

};

struct Adjvex

{

	int v;

	int value;

};

int root[101],wight[101],visit[101];

node ss[10001],tt[10001];

vector <Adjvex> map[101];

bool flag;

int cmp(const void* a,const void * b)

{

	return  ((node *)a)->value-((node*)b)->value;

}

int find(int i)

{

	while(root[i]!=i)

		i=root[i];

	return i;

}

int dfs(int s,int t,int w)//搜索环路径

{

	int j;

	visit[s]=1;

	if(s==t) return 1;

	for(j=0;j<map[s].size();j++)

		if(!visit[map[s][j].v] && dfs(map[s][j].v,t,w))

		{

			if(map[s][j].value==w) flag=true;

			return 1;

		}

	//visit[s]=0;

    return 0;

}

int Union(int s,int t)

{

    if(wight[s]>=wight[t])

	{

		root[t]=s;

		wight[s]+=wight[t];

	}

	else

	{

		root[s]=t;

		wight[t]+=wight[s];

	}

	return 0;

}

int add(node s)//构造MST的数据结构，搜索时用

{

    Adjvex e={s.end,s.value};

	map[s.start].push_back(e);

	e.v=s.start;

	map[s.end].push_back(e);

	return 0;

}

int main()

{

	int i,k,m,n,N,sum,s,t,count;

	cin>>n;

	while(n--)

	{

		cin>>m>>k;

		for(i=0;i<k;i++)

			cin>>ss[i].start>>ss[i].end>>ss[i].value;

        qsort(ss,k,sizeof(ss[0]),cmp);

        for(i=1;i<=m;i++)

		{ 

			root[i]=i; wight[i]=1;

		}

		sum=N=count=0;

		for(i=0;i<k;i++)

		{

			s=find(ss[i].start);

			t=find(ss[i].end);

			if(s==t)

			{ 

				tt[N++]=ss[i]; continue; 

			}

			Union(s,t);

			add(ss[i]);

			sum+=ss[i].value;

			//if(count++>=m-1) break;

		}

		for(flag=false,i=0;i<N;i++)

		{

			memset(visit,0,sizeof(visit));

			dfs(tt[i].start,tt[i].end,tt[i].value);

			if(flag) break;

		}

		if(flag)

			cout<<"Not Unique!"<<endl;

		else

			cout<<sum<<endl;

		for(i=0;i<=m;i++)

			map[i].clear();

	}

	return 0;

}