题意:有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
*/