hdu 5441 离线处理+并查集

参考

点击打开链接

#include <iostream>
#include <algorithm>
#include <vector>
#include <cstdio>
#include <cstring>
using namespace std;
const int M=2*1e6+20;
struct Query
{
    int x,id;
    bool operator < ( const Query& a )const 
    {
        return x < a.x;
    }
}a[M];
struct Edge{
	int u,v,w;
	bool operator<(const Edge& t)const{
		return w<t.w;
	}
}e[M];
int n,m,q;
int query[M];
long long res[M];
int fa[M],num[M];//num[i]集合i的个数 
int find(int x)
{
	if(fa[x]!=x)
	{
		fa[x]=find(fa[x]);
	}
	return fa[x];
}
void Union(int x,int y)
{
	
	fa[y]=x;
	num[x]+=num[y];
} 
int main()
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>n>>m>>q;
		for(int i=1;i<=n;i++)
		{
			fa[i]=i;
			num[i]=1;
		}
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
		}
		for(int i=1;i<=q;i++)
		{
			a[i].id=i;
			scanf("%d",&a[i].x);
		}
		sort(e+1,e+m+1);
		sort(a+1,a+q+1);//对要询问的边排序 离线处理 
	
		int cur=1;//已经加入的边 
		
		long long ans=0;
		for(int i=1;i<=q;i++)
		{
		//将路径中边权<=q[i]的点加入到集合中 num为集合元素的个数 
		//所以集合中任意两点路径上边权均不超过q[i]
		//个数为A(num,2) =num*(num-1)
			while(cur<=m&&e[cur].w<=a[i].x)
			{
				int u=find(e[cur].u);//加边并判断这条边的两个顶点是否已经在集合中 
				int v=find(e[cur].v);
				cur++;
				if(u==v) continue; 
				else
				{
				
					ans+=(num[u]+num[v])*(num[u]+num[v]-1)-num[u]*(num[u]-1)-num[v]*(num[v]-1);
					//现在个数-原来个数=新增个数 
					Union(u,v);
				}
			}
		
			res[a[i].id]=ans;//按题目所给顺序输出 
		}
		for(int i=1;i<=q;i++)
		{
			printf("%lld\n",res[i]);
		}
	}
	return 0;
}