【旅游】题解

题目

题目描述

输入

输出

样例输入

1
5 5 3
2 3 6334
1 5 15724
3 5 5705
4 3 12382
1 3 21726
6000
10000
13000

样例输出

2
6
12

数据制约

分析

先把路径和询问也按权值排序，依次按权值把边加入图中，则增加的边数就是边的两头所在的连通块的节点数的积再乘2（如果在同一个连通块就不用），再用并查集维护就可以了。

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <iostream>
int n,m,tot,q,test;
using namespace std;
int a[110000][4],bcj[21000],ans[6000],bl[21000],qu[6000][3];
//bl数组是以x为祖宗的连通块的节点数，bcj即并查集
void q1(int l,int r)
{
	int i=l,j=r,mid=a[(l+r)/2][3],e;
	while(i<j)
	{
		while(a[i][3]<mid) i++;
		while(a[j][3]>mid) j--;
		if(i<=j)
		{
			e=a[i][2];
			a[i][2]=a[j][2];
			a[j][2]=e;
			e=a[i][1];
			a[i][1]=a[j][1];
			a[j][1]=e;
			e=a[i][3];
			a[i][3]=a[j][3];
			a[j][3]=e;
			i++;
			j--;
		}
	}
	if(i<r) q1(i,r);
	if(l<j) q1(l,j);
}
void q2(int l,int r)
{
	int i=l,j=r,mid=qu[(l+r)/2][1],e;
	while(i<j)
	{
		while(qu[i][1]<mid) i++;
		while(qu[j][1]>mid) j--;
		if(i<=j)
		{
			e=qu[i][1];
			qu[i][1]=qu[j][1];
			qu[j][1]=e;
			e=qu[i][2];
			qu[i][2]=qu[j][2];
			qu[j][2]=e;
			i++;
			j--;
		}
	}
	if(i<r) q2(i,r);
	if(l<j) q2(l,j);
}
int zz(int x)
{
	if(x==bcj[x]) return x;
	bcj[x]=zz(bcj[x]);
	return bcj[x];
}
int main()
{
	cin>>test;
	int i,j,k,x,y,l;
	while(test--)
	{
		scanf("%d%d%d",&n,&m,&q);
		fill(qu[1],qu[1]+5600,0);
		fill(qu[2],qu[2]+5600,0);
		for(i=1;i<=n;i++)
		{
		    bcj[i]=i;
		    bl[i]=1;
		}
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&a[i][1],&a[i][2],&a[i][3]);
		}
		for(i=m+1;i<=100000;i++)
		    a[m+1][1]=a[m+1][2]=a[m+1][3]=200000000;
		q1(1,m);
		for(i=1;i<=q;i++)
		{
			scanf("%d",&qu[i][1]);
			qu[i][2]=i;
		}
		q2(1,q);
		j=1;
		int t=0;//t用来累加
		for(i=1;i<=q;i++)
		{
			while(qu[i][1]>=a[j][3] && j<m)
			{
				x=zz(a[j][1]);//找祖宗
				y=zz(a[j][2]);//找祖宗
				if(x!=y)
				{
					bcj[y]=x;
					t+=bl[x]*bl[y]*2;
					bl[x]+=bl[y];
					bl[y]=0;
				}
				j++;
			}
			ans[qu[i][2]]=t;
		}
		for(i=1;i<=q;i++)
		{
		    printf("%d
",ans[i]);
		}
	}
}