http://acm.sdut.edu.cn/sdutoj/problem.php?action=showproblem&problemid=2494
按求最小生成树的步骤 先按边的长度进行排序
我们把每一个边权相等的一个段看做一个单位
那我们一个单位一个单位的求
假如求到第k个单位了 无论前面k-1个单位怎么取舍 结果都是前面出现的所有端点都在一个联通块里面
无论度第k个单位怎么取舍前k个单位出现的端点也都在一个联通块里(求第k+1个联通块时)
所有按照排好的顺序 每个单位是相互独立的
对一到了第k个单位时 先把这个单位的所有边先搜一遍 如果哪个边的两个端点不在一个联通块内 则说明它是有资格成为某个最小生成树中的一员
所有计数 但不做其他操作
然后再搜一遍 把所以端点都联通
最后计数结果就是答案
代码:
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<string>
#include<queue>
#include<stack>
#include <iomanip>
using namespace std;
#define LL long long
const int INF=0x3f3f3f3f;
//priority_queue<int,vector<int>,greater<int> >qt;
const int N=100005;
const int M=1000005;
int f[N];
struct node
{
int l,r,k;
}mem[M];
bool cmp(node x,node y)
{
return x.k<y.k;
}
int findx(int x)
{
if(f[x]!=x)
f[x]=findx(f[x]);
return f[x];
}
int main()
{
//freopen("data.in","r",stdin);
int n,m;
while(cin>>n>>m)
{
for(int i=0;i<m;++i)
cin>>mem[i].l>>mem[i].r>>mem[i].k;
sort(mem,mem+m,cmp);
for(int i=1;i<=n;++i)
f[i]=i;
int sum=0;
int I;
for(int i=0;i<m;i=I)
{
for(I=i;I<m;++I)
{
if(mem[I].k!=mem[i].k)
break;
if(findx(mem[I].l)!=findx(mem[I].r))
++sum;
}
for(int j=i;j<I;++j)
{
f[findx(mem[j].l)]=findx(mem[j].r);
}
}
cout<<sum<<endl;
}
return 0;
}