修路方案 nyoj

这道题是最小生成树的题。可是我却找不到方法去比较两种最小成树的不同判断方法，本来是用想用并查集的集合元素的不同来储存不同的状态，可是想想却发现不可行，应为元素都是一样的。。。。。

 
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
#define maxN 510
#define MAX 0x0fffffff
#define MIN -0x0fffffff
int N,M,map[maxN][maxN],dis[maxN],maxlen[maxN][maxN],pre[maxN];
bool vis[maxN];
int prim()
{
	int i,j,k,minn,pr;
	memset(vis,false,sizeof(vis));
	for(i=1;i<=N;i++){dis[i]=map[1][i];pre[i]=1;}
	vis[1]=true;
	for(j=1;j<N;j++)
	{
		minn=MAX;
		for(i=1;i<=N;i++)
			if(!vis[i]&&dis[i]<minn)
			{
				k=i;
				minn=dis[i];
			}
		pr=pre[k]; 
		maxlen[k][pr]=maxlen[pr][k]=map[k][pr];
		for(i=1;i<=N;i++)
			if(vis[i])
				maxlen[i][k]=maxlen[k][i]=max(maxlen[i][pr],maxlen[k][pr]);
		vis[k]=true;
		for(i=1;i<=N;i++)
			if(!vis[i]&&dis[i]>map[i][k])
			{
				dis[i]=map[i][k];
				pre[i]=k;
			}
	}
	for(i=1;i<N;i++)
		for(j=i+1;j<=N;j++)
			if(pre[i]==j||pre[j]==i)continue;
			else if(maxlen[i][j]==map[i][j])return 1;
	return 0;
}
int main()
{
	//freopen("1.txt","r",stdin);
	int T;
	scanf("%d",&T);
	while(T--)
	{
		int u,v,w;
		scanf("%d%d",&N,&M);
		for(int i=0;i<=N;i++)
			for(int j=0;j<=N;j++)
			{
				map[i][j]=MAX;
				maxlen[i][j]=MIN;
			}
		for(int i=0;i<M;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			map[u][v]=map[v][u]=w;
		}
		if(prim())printf("Yes\n");
		else printf("No\n");	
	}
}