HDU 5441 Travel (带权并查集)

http://acm.hdu.edu.cn/showproblem.php?pid=5441

题意是一个无向图，每个点之间都有一个权值。然后有Q次询问，每次询问输入一个值x，如果两点之间的权值不大于x，那么该条路可走。对于每次询问输出一共有多少组走法。

开始想到的是用类似最短路的算法处理一下每两个点之间的距离，然后对每次询问都之间枚举两个点的权值是否小于x即可。但是点，边都特别大，显然不是好的方法。

后来看了巨巨的博客，才知道用并查集的思想处理。把边按从小到大的顺序先排个序，然后当边权小于输入的x的时候即可将该边的两点加入到集合中。处理的时候也要把输入的x从小到大排个序。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int N = 20000+10;
const int INF = 0x3f3f3f3f;
int father[N];
int rela[N];
int ans[5050];
int n,m,q,res;
struct edge
{
    int u,v,w;
    edge(int uu,int vv,int ww)
    {
       u=uu,v=vv,w=ww;
    }
    friend bool operator < (const edge a,const edge b)
    {
        return a.w < b.w;
    }
};
vector<edge>v;
struct Node
{
    int index,val;
    friend bool operator < (const Node a,const Node b)
    {
        return a.val < b.val;
    }
}que[5050];
void init()
{
    for(int i=1;i<=n;i++)
    {
        father[i] = i;
        rela[i] = 1;
    }
    res = 0;
    v.clear();
    return;
}
int find(int x)
{
    int y=x;
    while(y!=father[y]) y=father[y];
    while(x!=father[x])
    {
        int px = father[x];
        father[x] = y;
        x = px;
    }
    return y;
}
void Union(int x,int y)
{
    int fx = find(x);
    int fy = find(y);
    if(fx!=fy)
    {
        int numx = rela[fx];
        int numy = rela[fy];
        res += numx*numy*2;
        father[fx] = fy;
        rela[fy] += rela[fx]; 
    }
    return;
}
void solve()
{
    int tmp = 0;
    for(int i=1;i<=q;i++)
    {
        while(tmp < m && v[tmp].w <= que[i].val)
        {//直到所有的边都合并 或者 当前边权大于询问的x值
            Union(v[tmp].u,v[tmp].v);
            tmp++;
        }
        ans[que[i].index] = res;
    }
    return;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d %d",&n,&m,&q);
        init();
        for(int i=1;i<=m;i++)
        {
            int st,ed,w;
            scanf("%d %d %d",&st,&ed,&w);
            v.push_back(edge(st,ed,w));
        }
        sort(v.begin(),v.end());//将每条边都按权值大小排个序先
        for(int i=1;i<=q;i++)
        {
            que[i].index = i;
            scanf("%d",&que[i].val);
        }
        sort(que+1,que+1+q);//将输入要询问的值也排个序
        solve();
        for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
    }
    return 0;
}