POJ 1679

求最小生成树是否唯一，即求次小生成树，判断它的边权和是否和最小生成树的一样。

还是用的kruskal算法，先求得最小生成树，再记录下路径，然后枚举删除最小生成树中的一条边，再构造新的生成树，看是否和最小权值相同。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
#define maxx 105
int n,p[maxx],m;
struct Edge
{
  int u,v,w;
}edge[maxx*maxx/2];
bool cmp(Edge a,Edge b)
{
  return a.w<b.w;
}
int findpa(int x)
{
  return p[x]==x?x:p[x]=findpa(p[x]);
}
int kruskal()
{
  for(int i=1;i<=n;i++)
  p[i]=i;
  int sum=0,num=0;
  int path[maxx*maxx];
  sort(edge,edge+m,cmp);
  for(int i=0;i<m;i++)
  {
    int p1=findpa(edge[i].u);
    int p2=findpa(edge[i].v);
    if(p1!=p2)
	{
	  sum+=edge[i].w;
	  p[p1]=p2;
	  path[num++]=i;
	}
  }
  for(int i=0;i<num;i++)  //依次删掉每条边，再建树
  {
    for(int k=1;k<=n;k++)
	p[k]=k;
	int sum2=0,num2=0;
	for(int j=0;j<m;j++) //再枚举剩余的边
	{
	  if(path[i]==j) continue;
	  int p1=findpa(edge[j].u);
	  int p2=findpa(edge[j].v);
	  if(p1!=p2)
	  {
	    num2++;
	    sum2+=edge[j].w;
	    p[p1]=p2;
	  }
	}
	if(num2==n-1&&sum2==sum)
	return -1;
  }
  return sum;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
      scanf("%d%d",&n,&m);
	  for(int i=0;i<m;i++)
	  {
	  	scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
	  }
	  int ans=kruskal();
	  if(ans==-1)
	  printf("Not Unique!\n");
	  else
	  printf("%d\n",ans);
	}
	return 0;
}