FZU 2087 统计树边

题目链接：http://acm.fzu.edu.cn/problem.php?pid=2087

题意：给出一个有n个点m条边的图，没有重边和自环，问有多少条边可以是最少生成树上的边

思路：对Kruscal不熟悉……没有想出来，图上只有2种边，一种是最少生成树上的边，一种不是最少生成树上的边，每次贪心的时候有权值相同且不在同一个集合的边res++，之后将这些边全放进生成树的集合里，最终得到的答案就是所求

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define maxn 100030
using namespace std;

struct edge
{
int x,y,w;
edge(int x=0,int y=0,int w=0):x(x),y(y),w(w){}
}e[maxn];

int fa[maxn];
int n,m;

bool cmp(edge p,edge q)
{
return p.w<q.w;
}

int finds(int x)
{
if (x==fa[x]) return x;
else return fa[x]=finds(fa[x]);
}

int Kruscal()
{

sort(e+1,e+m+1,cmp);
int cnt=n;
int res=0;
for (int i=0;i<=n;i++)
fa[i]=i;
for (int i=1;i<=m;i++)
{
int pos=i;
while (e[i].w==e[pos].w)
{
int t1=finds(e[pos].x);
int t2=finds(e[pos].y);
if (t1!=t2)
res++;
pos++;
}

pos=i;
while (e[i].w==e[pos].w)
{
int t1=finds(e[pos].x);
int t2=finds(e[pos].y);
if (t1!=t2)
fa[t1]=t2;
pos++;
}
}
return res;
}

int main()
{
int t;
scanf("%d",&t);
while (t--)
{
scanf("%d%d",&n,&m);
for (int i=1;i<=m;i++)
{
scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
}
int res=Kruscal();
printf("%d\n",res);
}
}