nssl 1486.旅游

$Description$

给定一张无向图$G\{n,m\}$，每条边有一个权值$d_i$
有$q$次询问，每次给定一个$x$，规定所有$d_i\le x$的是可以走的，求有多少点对$(i,j)$满足$i$可以走到$j$

数据范围：$n\le 2\times 10^4,m\le 10^5,q\le 5\times 10^3$

---

$Solution$

首先对$x$排序，再对$d_i$排序，问题就转化成为依次开放一些边，求连通块内路径数
用并查集维护即可

时间复杂度：$O(mlogm+qloqq\alpha (n)+n\alpha (n))$

---

$Code$

#include
#include
#include
#define LL long long
using namespace std;int n,m,q,T,f[20010],siz[20010],ans[5010],nowsum;
struct node{int from,to,w;}e[100010];
struct data{int id,x;}qs[5011];
inline bool cmp(node x,node y){return x.w<y.w;}
inline bool cop(data x,data y){return x.x<y.x;}
inline int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
inline LL read()
{
	char c;LL d=1,f=0;
	while(c=getchar(),!isdigit(c)) if(c=='-') d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),isdigit(c)) f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	T=read();
	while(T--)
	{
		n=read();m=read();q=read();
		for(register int i=1;i<=n;i++) f[i]=i,siz[i]=1;
		for(register int i=1;i<=m;i++) e[i].from=read(),e[i].to=read(),e[i].w=read();
		for(register int i=1;i<=q;i++) qs[i].x=read(),qs[i].id=i;
		sort(e+1,e+1+m,cmp);
		sort(qs+1,qs+1+q,cop);
		nowsum=0;
		for(register int i=1,j=0;i<=q;i++)
		{
			while(e[j+1].w<=qs[i].x&&j<m)
			{
				j++;
				int fx=find(e[j].from),fy=find(e[j].to);
				if(fx==fy) continue;
				nowsum+=siz[fx]*siz[fy];
				if(fx<fy) f[fy]=fx,siz[fx]+=siz[fy];
				else f[fx]=fy,siz[fy]+=siz[fx];
			}
			ans[qs[i].id]=nowsum;
		}
		for(register int i=1;i<=q;i++) printf("%d\n",ans[i]*2);
	}
}