hdu 3938 Portal(离线并查集)

题目：

        链接：点击打开链接

题意：

        更大的香蕉需要更大的香蕉皮。不幸的是做一对大门需要至少消耗T能量，T是U和V之间最长边的值。两点之间可能有很多条路径。有L能量值，问能做多少路。

思路：

        边和询问都从小到大排序。

        对于i询问，结果是询问i-1的结果加上询问i-1两点之间的边合并的值。

        对一条两点之间的边，如果U和V不同集合，那么新路就是U所在集合*V所在集合。

代码：

 

       

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define MAXN 10010
#define MAXM 50050

struct node{
    int u,v,w;
}edge[MAXM];

struct query{
    int num,val;
}qu[MAXN];

int n,m,q;
int g[MAXN],root[MAXN];
int cnt[MAXN];

int cmp_e(node a,node b)
{
    return a.w < b.w;
}

int cmp_q(query a,query b)
{
    return a.val < b.val;
}

int findset(int x)
{
    return root[x] == x ? x : root[x] = findset(root[x]);
}

void mergeset(int x,int y)
{
    if(g[x] > g[y])
    {
        root[y] = x;
        g[x] += g[y];
    }
    else
    {
        root[x] = y;
        g[y] += g[x];
    }
}

void init()
{
    for(int i=1; i<=n; i++)
    {
        root[i] = i;
        g[i] = 1;
    }
}

int main()
{
    //freopen("input.txt","r",stdin);
    while(scanf("%d%d%d",&n,&m,&q) != EOF)
    {
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        }
        init();
        for(int i=1; i<=q; i++)
        {
            scanf("%d",&qu[i].val);
            qu[i].num = i;
        }
        sort(edge,edge+m,cmp_e);
        sort(qu+1,qu+q+1,cmp_q);
        int j = 0;
        for(int i=1; i<=q; i++)
        {
            cnt[qu[i].num] = cnt[qu[i-1].num];
            while(edge[j].w <= qu[i].val && j<m)
            {
                int x = findset(edge[j].u);
                int y = findset(edge[j].v);
                if(x != y)
                {
                    cnt[qu[i].num] += g[x] * g[y];
                    mergeset(x,y);
                }
                j++;
            }
        }
        for(int i=1; i<=q; i++)
            printf("%d\n",cnt[i]);
    }
    return 0;
}

--------------------------------------------------------------------

战斗，从不退缩；奋斗，永不停歇~~~~~~~~~~~~~~