USACO 2021 December Contest, SilverProblem 2. Connecting Two Barns

最新推荐文章于 2025-05-02 13:41:05 发布

weixin_67198346

最新推荐文章于 2025-05-02 13:41:05 发布

阅读量239

点赞数

文章标签：算法 c++

本文链接：https://blog.csdn.net/weixin_67198346/article/details/132072088

版权

题意：有n块田地，m条道路，有道路相连的两块田地可相互到达，现在可以添加0、1或2条道路，使得添加完的图田地1可到达田地n,添加一条道路所需要花费的值为(i-j)^2,求所需花费的最小值。

思路：这是一道考察并查集和二分查找的题目，先用并查集将每个点找到父节点，分为k个集团，并把能到田地1 和田地n的点分别放入集合p和q中，再将k个集团中每个集团到p和q所需要的最小花费记作数组f和ff，在遍历田地1到田地n中的田地i，二分其在集合p和q中的位置，计算 i 连接集合p和q所需要最小的花费，更新f和ff的最小值，最后分别计算只连1、2条边所需的最小值（若只用连0条边fa[1]==fa[n]），代码如下

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll aa,a,b,c,d,fa[100005],tmp,f[100005],ff[100005],p[100005],q[100050];
ll find(ll x) {
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}
inline int read() {
	int x = 0, f = 1;
	char ch = getchar();
	for (; !isdigit(ch); ch = getchar()) if (ch == '-') f = -f;
	for (; isdigit(ch); ch = getchar()) x = (x << 1) + (x << 3) + (ch ^ 48);
	return x * f;
}
int main(){
	aa=read();
	while(aa--){
		a=read();
		b=read();
		for(int i=1;i<=a;i++){
			fa[i]=i;
			f[i]=100000000000000000;
			ff[i]=100000000000000000;
			p[i]=0;
			q[i]=0;
		}
		for(int i=1;i<=b;i++){
			c=read(),d=read();
			ll fc=find(c);
			ll fd=find(d);
			if(fc!=fd)fa[fc]=fd;
		}
		ll f1=find(1),fn=find(a),tmp1=0,tmp2=0;
		if(f1==fn){
			cout<<"0\n";
			continue;
		}
		for(int i=1;i<=a;i++){
			ll fi=find(i);
			if(fi==f1){
				p[++tmp1]=i;
			}			 
			else if(fi==fn){
				q[++tmp2]=i;
			}
		}
		f[f1]=0,ff[fn]=0;
		for(int i=1;i<=a;i++){
			ll t;
			ll fi=find(i);
			if(fi!=f1){
				ll pre=upper_bound(p+1,p+1+tmp1,i)-p-1;
				f[fi]=min(f[fi],(i-p[pre])*(i-p[pre]));
				if(pre<tmp1){
					f[fi]=min(f[fi],(i-p[pre+1])*(i-p[pre+1]));
				}				
			} 
			if(fi!=fn){
				ll r1=upper_bound(q+1,q+1+tmp2,i)-q;
				ff[fi]=min(ff[fi],(i-q[r1])*(i-q[r1]));					
				if(r1>1){
					ff[fi]=min(ff[fi],(i-q[r1-1])*(i-q[r1-1]));
				}
			}	
		}										
		long long ans=0x3f3f3f3f3f3f3f3f;
		for(int i=1;i<=a;i++){
			ll fi=find(i);
			ans=min(f[fi]+ff[fi],ans);
		} 
		printf("%lld\n",ans);
	}
} 
/*
1
10
5
1 2 2 7
4 6 5 9  9 10
*/