这题题目意思没看懂,看了题解之后才明白,写一下我的理解
题意:一个无向图,给q个询问,每个询问一个l值,询问边的大小限定在l以内,能生成路径的条数。可以理解为限定边的最大长度后,仍有多少对点之间还是连着的。有个隐藏条件:不能成环,不然可以构成无数条路径。考虑kruskal过程,边加入集合中的顺序是从小到大加入,最大边是限定的。没加边前每个集合都是单个结点,结点数量为1,集合合并,合并后的集合路径数会增加A集合的路径数*B集合的路径数。
#include<bits\stdc++.h>
using namespace std;
const int maxe=5*1e4+10;
const int maxn=1e4+10;
int f[maxn],num[maxn],ans[maxn];
struct edge
{
int u,v,w;
}e[maxe];
struct que
{
int id;
int l;
}q[maxn];
int Find(int x)
{
return f[x]==x?x:f[x]=Find(f[x]);
}
bool cmp1(edge a,edge b)
{
return a.w<b.w;
}
bool cmp2(que a,que b)
{
return a.l<b.l;
}
void init(int n)
{
memset(ans,0,sizeof(ans));
for(int i=1;i<=n;i++)f[i]=i,num[i]=1;
}
int merge(int u,int v)
{
int xx=Find(u);
int yy=Find(v);
if(xx!=yy)
{
f[xx]=yy;
int temp=num[yy]*num[xx];
num[yy]+=num[xx];
return temp;
}
else
return 0;
}
int main()
{
int n,m,Q;
// freopen("in.txt","r",stdin);
// freopen("out1.txt","w",stdout);
while(~scanf("%d%d%d",&n,&m,&Q))
{
int i;
for(i=0;i<m;i++)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
for(i=0;i<Q;i++)
{
scanf("%d",&q[i].l);
q[i].id=i;
}
sort(e,e+m,cmp1);
sort(q,q+Q,cmp2);
//集合初始化
init(n);
int cnt=0,sum=0;
for(i=0;i<Q;i++)//小的和小的先比较
{
while(q[i].l>=e[cnt].w&&cnt<m)
{
int u=e[cnt].u,v=e[cnt].v;
sum+=merge(u,v);
cnt++;
}
ans[q[i].id]=sum;
}
for(i=0;i<Q;i++)
printf("%d\n",ans[i]);
}
}