hdu 3938 (离线并查集)

#include <stdio.h>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <iostream>
#define pi acos(-1.0)
#define INF 0x3f3f3f3f
using namespace std;
#define ll long long
const int maxn=50010;
int n,m,q;
int f[maxn],num[maxn];
struct edge
{
    int u,v,w;
    bool operator < (const edge tmp) const
    {
        return w<tmp.w;
    }
}e[maxn];
struct node
{
    int id,l;
    bool operator < (const node tmp) const
    {
        return l<tmp.l;
    }
}nd[maxn];
int found(int x)
{
    if(x!=f[x]) f[x]=found(f[x]);
    return f[x];
}
int Merge(int a,int b)
{
    int fx=found(a),fy=found(b);
    if(fx==fy) return 0;  //在集合中，已经计算过了
    f[fx]=fy;
    int ans=num[fx]*num[fy];
    num[fy]+=num[fx];
    num[fx]=0;
    return ans;
}
int main()
{
    //freopen("C:\\Users\\Administrator\\Desktop\\a.txt","r",stdin);
    //ios::sync_with_stdio(false);
    //freopen("C:\\Users\\Administrator\\Desktop\\b.txt","w",stdout);
    int ans[maxn];
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        memset(ans,0,sizeof ans);
        for(int i=0;i<m;i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        for(int i=0;i<q;i++)
        {
            scanf("%d",&nd[i].l);
            nd[i].id=i;
        }
        sort(e,e+m);
        sort(nd,nd+q);
        for(int i=0;i<=n;i++)
            num[i]=1,f[i]=i;
        int pos=0,tmp=0;
        for(int i=0;i<q;i++)
        {
            while(pos<m&&nd[i].l>=e[pos].w)
            {
                tmp+=Merge(e[pos].u,e[pos].v);
                pos++;
            }
            ans[nd[i].id]=tmp;
        }
        for(int i=0;i<q;i++)
            printf("%d\n",ans[i]);
    }
    return 0;
}