题目大意:
有一个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]);
}
}
}