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;}