poj3938 Portal 并查集

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

题目大意:给你n个点，然后输进去m条x到y的路径，长度为v，再然后有q次询问，每次询问一条路径中最长的路的长度不超过第i次询问的长度L的有多少条？

 思路：先把路径长度按照升序排列，再把长度L按照升序排列，然后依次从小到大遍历V，如果V<=L，则把当前点合并，这里的rank[i]表示第i个元素所在集合的结点个数，故两个集合合并后产生的总路径条数为rank[i]*rankj]，由于是把所有条件都给出来了，所以这个过程是离线算法，相当于预处理，代码如下：

#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 10005
using namespace std;
struct node
{
    int x;
    int y;
    int v;
} s[5*N];
struct point
{
    int z,id;
} s1[N];
int solve[N],f[N],rank[N],n,m,q;
void init()//预处理
{
    for(int i=1; i<=n; i++)
    {
        f[i]=i;
        rank[i]=1;
    }
}
bool cmp(node a,node b)
{
    return a.v<b.v;
}
bool cmp1(point a,point b)
{
    return a.z<b.z;
}

int find(int x)
{
    if(x==f[x]) return x;
    f[x]=find(f[x]);
    return f[x];
}
int merge(int x,int y)
{
    int ans;
    int fx=find(x);
    int fy=find(y);
    if(fx==fy) return 0;
    ans=rank[fx]*rank[fy];//两个集合合并产生ans条路径
    rank[fx]+=rank[fy];
    f[fy]=fx;
    return ans;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&q))
    {
        memset(solve,0,sizeof(solve));
        for(int i=0; i<m; i++)
            scanf("%d%d%d",&s[i].x,&s[i].y,&s[i].v);
        sort(s,s+m,cmp);
        for(int i=0; i<q; i++)
        {
            scanf("%d",&s1[i].z);
            s1[i].id=i;
        }
        sort(s1,s1+q,cmp1);
        init();
        int j=0,sum=0;
        for(int i=0; i<q; i++)
        {
            while(j<m&&s1[i].z>=s[j].v)
            {
                sum+=merge(s[j].x,s[j].y);
                j++;
            }
            solve[s1[i].id]=sum;
        }
        for(int i=0; i<q; i++)
            printf("%d\n",solve[i]);
    }
}

对于最后的for循环，我感觉类似于贪心思想，即先把路径长度小的和小L比较，如果比L小，就合并，否则找更大的L，这样肯能保证有最多路径。。。