hdu 5441 最小生成树

题目大意：
有一个n个点的无向图，给出m条边的边权，给出q次询问，每次给出一个值，求用到所有边权不大于这个值的边的情况下，能够互相到达的点对的个数（自己到自己不算）

#include<bits/stdc++.h>
using namespace std;
int head[40000],top;
int f[40000];
long long r[40000];
int find(int x)
{
    if(x==f[x])
        return x;
        else return f[x]=find(f[x]);
}
struct node
{
    int u,v,val;
   /*bool operator < (const node & a) const
   {
       val<a.val;
   }*/
}e[400000];
bool cmp(node a,node b)
{
    return a.val<b.val;
}
int ans[40000];
long long l[40000];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].val);
        }
        for(int i=0;i<=n;i++)
        {
            f[i]=i;
            r[i]=1;
        }
        sort(e,e+m,cmp);
        int p=2;
        int sum=n-1;
        for(int i=0;i<m;i++)
        {
            int x=find(e[i].u);
            int y=find(e[i].v);
           // printf("%d.........\n",e[i].val);
            if(x!=y)
            {
                f[x]=y;
                ans[p]=e[i].val;
                l[p++]=2*r[y]*r[x];
                r[y]+=r[x];
                sum--;
                if(!sum)
                    break;
            }
        }
        l[0]=ans[0]=l[1]=ans[1]=0;
        for(int i=1;i<p;i++)
        {
            l[i]+=l[i-1];
        }
        for(int i=0;i<k;i++)
        {
            int x;
            scanf("%d",&x);
            int pos=upper_bound(ans,ans+p,x)-ans-1;
            printf("%lld\n",l[pos]);
        }
    }
}