Codeforces Round #542 [Alex Lopashev Thanks-Round] (Div. 2) C. Connect

C. Connect

题目链接-C. Connect



题目大意
给出起点终点，然后给你一张图，0代表陆地 1代表水，可以上下左右走陆地，也可以在两块陆地之前建隧道走，问你从起点到终点需要建的隧道最短长度是多少

解题思路
$dfs$

• 先对起点$(a,b)$深搜判断与终点$(c,d)$是否属于一个连通块，同时记录与起点属于一个连通块的点的坐标
• 若vis[c][d]==1，则说明起点和终点属于一个连通块，即无需建隧道，输出0即可
• 若起点和终点不属于一个连通块，则对终点也进行深搜，记录与终点属于一个连通块的点的坐标
• 因为我们可以在任意两个陆地之前建隧道，且题目数据比较小，所以我们可以遍历起点连通块的所有点到终点连通块的所有点的距离，并求出最小值输出即可
• 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e3+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
char mp[60][60];
int n,a,b,c,d,vis[60][60],cnt;
struct node{
	int x,y;
}s[N],t[N];
void dfs(int x,int y,node *f){
	for(int i=0;i<4;i++){
		int dx=x+dir[i][0];
		int dy=y+dir[i][1];
		if(dx<1||dy<1||dx>n||dy>n||vis[dx][dy]||mp[dx][dy]=='1')
			continue;
		vis[dx][dy]=1;
		f[cnt].x=dx,f[cnt++].y=dy;
		dfs(dx,dy,f);
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	cnt=1;
	cin>>n>>a>>b>>c>>d;
	s[0].x=a,s[0].y=b;
	t[0].x=c,t[0].y=d;
	vis[a][b]=1;
	for(int i=1;i<=n;i++)
		cin>>mp[i]+1;
	dfs(a,b,s);
	if(vis[c][d]){
		cout<<0<<endl;
		return 0;
	}
	int num=cnt;
	cnt=1;
	vis[c][d]=1;
	dfs(c,d,t);
	int minx=INF;
	for(int i=0;i<num;i++){
		for(int j=0;j<cnt;j++){
			int tmp=(s[i].x-t[j].x)*(s[i].x-t[j].x)+(s[i].y-t[j].y)*(s[i].y-t[j].y);
			minx=min(minx,tmp);
		}
	}
	cout<<minx<<endl;
	return 0;
}