HDU 3938 离线并查集

/*
离线并查集，边按小到大排序，询问也按小到大排序；
对于询问x,答案就是询问x-1的值加上询问x-1的L1和询问x的L2之间的边合并带来的值
对于一条属于L1和L2之间的边，如果端点u，v在一个集合中，忽略，否则新开的路是u
所在集合的大小乘上v所在集合的大小
*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#include<vector>
#include<stack>
#include<map>
using namespace std;
const int M=50009;
const int N=10009;
struct Edge
{
    int u,v,w;
}edge[M];
bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}
struct Que
{
    int num,val;
}Qu[N];
bool cmp2(Que a,Que b)
{
    return a.val<b.val;
}
int n,m,q;
int fa[N],heav[N];
int ans[N];
void init()
{
    for(int i=0;i<m;i++)
    {
        scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
    }
}
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int Union(int x,int y)
{
    if(heav[x]>heav[y])
    {
        fa[y]=x;
        heav[x]+=heav[y];
    }
    else
    {
        fa[x]=y;
        heav[y]+=heav[x];
    }
}
void solve()
{
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
        heav[i]=1;
    }
    for(int i=1;i<=q;i++)
    {
        scanf("%d",&Qu[i].val);
        Qu[i].num=i;
    }
    sort(edge,edge+m,cmp);
    sort(Qu+1,Qu+q+1,cmp2);
    int j=0;
    for(int i=1;i<=q;i++)
    {
        ans[Qu[i].num]=ans[Qu[i-1].num];
        while(edge[j].w<=Qu[i].val&&j<m)
        {
            int x=find(edge[j].u);
            int y=find(edge[j].v);
            if(x!=y)
            {
                ans[Qu[i].num]+=heav[x]*heav[y];
              //  cout<<Qu[i].num<<" # "<<ans[Qu[i].num]<<endl;
                Union(x,y);
            }
            j++;
        }
    }
    for(int i=1;i<=q;i++)
    {
        printf("%d\n",ans[i]);
    }
}
int main()
{
    while(scanf("%d%d%d",&n,&m,&q)!=EOF)
    {
        init();
        solve();
    }
return 0;
}