hdu 4750——Count The Pairs

题意：南京理工大学有很多景点。两个景点组成一对。两个景点的价值用这种方式计算：设两个景点之间一条路径中最长的一条路的距离是s，这两个景点的价值f就是所有路径中s的最小值。游客想要找价值大于等于t的景点对有多少个。
思路：将边按照权值从小到大排序。依次遍历每条边。因为每次加入的边是当前遍历过的边中最大的，若这条边连接的两个节点不在同一个集合中，那么这两个集合之间点对的价值就是这条边的权值。
错误：合并的时候把两个集合大小计算写错了。。不是一次犯这种低级错误了。
代码如下：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<cstring>
#include<map>
#include<set>
#include<list>
#include<stack>
#include<algorithm>
#include<queue>
#include<vector>
#include<time.h>
#include<iomanip>
#include<assert.h>
using namespace std;

typedef long long ll;

struct Edge
{
    int u,v,w;
    bool operator < (const Edge &a) const {
        return w < a.w;
    }
    bool operator == (const Edge &a) const {
        return w == a.w;
    }
    bool operator > (const Edge &a) const {
        return w>a.w;
    }
    Edge(int u=0,int v=0,int w=0):u(u),v(v),w(w){}
}edge[500005];

int fa[10005];
int cnt[10005];

int find(int x)
{
    return fa[x] = fa[x] == x? fa[x] : find(fa[x]);
}

void merge(int x,int y)
{
    int a = find(x);int b = find(y);
    fa[a] = b;
    cnt[b] += cnt[a];
    cnt[a] += cnt[b];
}

int n,m;
ll ans[500005];

int bisearch(int x){
    int l = 0;int r = m-1;
    while(l<=r){
        int mid = (l+r);mid>>=1;
        if(edge[mid].w == x){
            return mid;
        }
        if(edge[mid].w<x){
            l = mid+1;
        } else r = mid-1;
    }
    return l;
}

int main()
{
//    freopen("data.txt","r",stdin);
    while(scanf("%d%d",&n,&m)!=EOF){
        for(int i = 0 ; i < m; ++ i ){
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        }
        sort(edge,edge+m);
        for(int i = 0; i < n;++i){
            fa[i] = i;
            cnt[i] = 1;
        }
        memset(ans,0,sizeof(ans));
        for(int i = 0 ; i < m; ++i){
            int u = edge[i].u;
            int v = edge[i].v;
            int a = find(u);int b = find(v);
            if(a == b)continue;
            ans[i] = (ll)cnt[a]*(ll)cnt[b]*2;
            merge(a,b);
        }
        for(int i = m-2;i>=0;--i){
            ans[i] += ans[i+1];
        }
        int p;
        scanf("%d",&p);
        while(p--){
            int t;
            scanf("%d",&t);
            int c = bisearch(t);
            printf("%I64d\n",ans[c]);
        }
    }
    return 0;
}