fzu2087 统计树边 最小生成树应用

采用桶排序中的分类思想，将相同权值的边分成一类。
之后分别考虑不同权值的边，如果两个点所在集合不同（已最小生成树的点集和不在最小生成树上的点集），说明这条边至少是一个最小生成树上的边，然后先不要把边合并，继续考虑权值相同的边，直到没有了，再将点合并。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
const int MAXN=100000+5;
const int MAXM=100000+5;
int bin[MAXN];
int use[MAXM];
vector<int>vt[1024];

struct Edge
{
    int u,v,w;
} edges[MAXM];
int tol;

void addedge(int u,int v,int w)
{
    edges[tol].u=u;
    edges[tol].v=v;
    vt[w].push_back(tol);
    edges[tol++].w=w;
}

int Find(int x)
{
    if(bin[x]==-1)return x;
    else return bin[x]=Find(bin[x]);
}

int main()
{
    int _,n,m,i,j,u,v,w,ans=0;
    scanf("%d",&_);
    while(_--)
    {
        tol=0;
        for(i=0; i<=500; i++)
            vt[i].clear();
        scanf("%d%d",&n,&m);
        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            addedge(u,v,w);
        }
        memset(bin,-1,sizeof(bin));
        memset(use,0,sizeof(use));
        ans=0;
        for(i=1; i<=500; i++)
        {
            if(vt[i].size()==0) continue;
            for(j=0; j<vt[i].size(); j++)
            {
                int t=vt[i][j];
                int u=edges[t].u;
                int v=edges[t].v;
                int t1=Find(u);
                int t2=Find(v);
                if(t1!=t2)
                {
                    ans++;
                    use[t]=1;
                }
            }
            for(j=0; j<vt[i].size(); j++)
            {
                int t=vt[i][j];
                if(!use[t]) continue;
                int u=edges[t].u;
                int v=edges[t].v;
                int t1=Find(u);
                int t2=Find(v);
                if(t1!=t2)
                {
                   bin[t1]=t2;
                }
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}