HDU5441 Travel 并查集

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5441

题目大意：n个顶点m条边的有向图G代表n个城市和之间的m条道路，边权w代表走过该边需要的时间，每个顶点都可以休息，给出q次询问，每次询问给出一个时间限制x，问你在每次连续行走都不超过x的时间段内，最多有多少个可以到达的城市对。(u,v)和(v,u)属于不同的城市对，当且仅当u!=v.

分析：我们首先把这些边按权值升序排序，然后对于每次询问，我们把按x升序排序，这样对于后面的询问，我们便可以由前面已询问的状态得出。具体做法就是：对于每次询问，从前到后遍历每一条边，直至当前边的权值大于该次询问的x值，此时已遍历的顶点构成一个连通分量，我们用并查集纪录每个连通分量中顶点的个数ran，那么很显然，该连通分量中满足条件的城市对的数目即为ran*(ran-1)，然后对于两个连通分量的合并，结果很显然为：(ran[u]+ran[v])*(ran[u]+ran[v]-1)-ran[u]*(ran[u]-1)-ran[v]*(ran[v]-1)。

实现代码如下：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=20005;
struct EDGE
{
    int u,v,w;
    bool operator <(const EDGE &a) const
    {
        return w<a.w;
    }
}edge[maxn*5];
struct QUERY
{
    int x,id;
    bool operator <(const QUERY &a) const
    {
        return x<a.x;
    }
}query[maxn];
int num[maxn];
int par[maxn],ran[maxn];
void Init(int n)
{
    for(int i=1;i<=n;i++)
    {
        par[i]=i;
        ran[i]=1;
    }
}
int Find(int x)
{
    if(par[x]!=x) return par[x]=Find(par[x]);
    return x;
}
void Union(int x,int y)
{
    par[y]=x;
    ran[x]+=ran[y];
}
int main()
{
    int t,n,m,q;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&q);
        Init(n);
        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<q;i++)
        {
            query[i].id=i;
            scanf("%d",&query[i].x);
        }
        sort(query,query+q);
        int cnt=0,ans=0;
        for(int i=0;i<q;i++)
        {
            while(cnt<m&&edge[cnt].w<=query[i].x)
            {
                int u=Find(edge[cnt].u);
                int v=Find(edge[cnt].v);
                cnt++;
                if(u==v) continue;
                ans+=(ran[u]+ran[v])*(ran[u]+ran[v]-1)-ran[u]*(ran[u]-1)-ran[v]*(ran[v]-1);
                Union(u,v);
            }
            num[ query[i].id ]=ans;
        }
        for(int i=0;i<q;i++)
          printf("%d\n",num[i]);
    }
    return 0;
}