湖南大学第十六届程序设计竞赛L题（dfs)

Cracked Pipes

题意：有如下六种水管编号1~6，输入一个nm大小的矩阵，表示这nm个水管的编号及位置，判断能否使水从（1，0）的位置流到（n，m+1）的位置

示例1
输入
2 3
1 4 6
5 3 2
输出
NO

示例2
输入
2 3
2 6 4
1 5 2
输出
YES

代码

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std; 
#define ll long long
#define INF 0x7fffffff
#define N 500010
 
int a[N],n,m;
int dic[5][2]={{-1,0},{0,-1},{1,0},{0,1}};
//mp数组判断该方向是否可连接
int mp[10][4]={{0,0,0,0},{1,0,1,0},{0,1,0,1},{0,0,1,1},{1,1,0,0},{1,1,0,1},{0,1,1,1} };
int vis[N]={0};
int getid(int a,int b){
	return m*(a-1)+b;
}
bool check(int x,int y){
	if(x<1||x>n||y<1||y>m)return false;
	if(vis[getid(x,y)])return false;
	else return true;
}
bool dfs(int x,int y){
	vis[getid(x,y)]=1;
	if(x==n&&y==m)return true;
	for(int i=0;i<4;i++){
		int fx=x+dic[i][0],fy=y+dic[i][1];
        //(i+2)%4表示与i相反的方向
		if(check(fx,fy)&&mp[a[getid(x,y)]][i]&&mp[a[getid(fx,fy)]][(i+2)%4])
			if(dfs(fx,fy))return true;
	}
	return false;
}
int main(){
	cin>>n>>m;
	int cnt=1;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			cin>>a[cnt++];
	if(mp[a[1]][1]==0||mp[a[getid(n,m)]][3]==0){
		cout<<"NO"<<endl;
		return 0;
	}
	if(dfs(1,1))cout<<"YES";
	else cout<<"NO";
}