HDU 5441 Travel(最短路径变形）

/*题意：有一个n个点的无向图，给出m条边的边权，给出q次询问，每次给出一个值，求用到所有边权不大于这个值的边的情况下，能够互相到达的点对的个数（自己到自己不算）
分析：
先把路线按照权值从小到大排序，然后再把那几个值按从小到大排序。后面就是一系列最短路径问题了，貌似不难*/
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxm=1e5;
int n,m,k;
int num[maxm];
int p[maxm];
int ans[maxm];
void Init()
{
    for(int i=0;i<=maxm;i++)
    {
        p[i]=i;
        num[i]=1;
    }
}
struct node
{
    int u,v,c;
    bool operator<(const node &s)const
    {
        return c<s.c;
    }
}t[maxm];
struct NODE
{
    int w,id;
    bool operator<(const NODE &p)const
    {
        return w<p.w;
    }

}T[maxm];
int find(int k)
{
    if(p[k]!=k)
    {
        return p[k]=find(p[k]);
    }
    else
    {
        return p[k];
    }
}
int main()
{
    int y;
    scanf("%d",&y);
    while(y--)
    {
        Init();
        memset(ans,0,sizeof(ans));
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&t[i].u,&t[i].v,&t[i].c);
        }
        sort(t,t+m);
        for(int i=0;i<k;i++)
        {
            scanf("%d",&T[i].w);
            T[i].id=i;
        }
        sort(T,T+k);
        int j=0;
        int cnt=0;
        for(int i=0;i<k;i++)
        {
            while(j<m&&t[j].c<=T[i].w)
            {
                int x=find(t[j].u);
                int y=find(t[j].v);
                j++;
                if(x!=y)
                {
                    p[x]=y;
                    cnt+=(num[x]*num[y]);
                    num[y]+=num[x];
                }
            }
            ans[T[i].id]=2*cnt;
        }
        for(int i=0;i<k;i++)
        {
            printf("%d\n",ans[i]);
        }
    }
    return 0;
}