HDU-3938 Portal 克鲁斯卡尔思想 (离线并查集)

这题题目意思没看懂,看了题解之后才明白,写一下我的理解
题意:一个无向图,给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]);
	}
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值