蓝桥杯历届试题 穿越雷区(dfs)

X星的坦克战车很奇怪,它必须交替地穿越正能量辐射区和负能量辐射区才能保持正常运转,否则将报废。
某坦克需要从A区到B区去(A,B区本身是安全区,没有正能量或负能量特征),怎样走才能路径最短?

已知的地图是一个方阵,上面用字母标出了A,B区,其它区都标了正号或负号分别表示正负能量辐射区。
例如:
A + - + -
- + - - +
- + + + -
+ - + - +
B + - + -

坦克车只能水平或垂直方向上移动到相邻的区。

数据格式要求:

输入第一行是一个整数n,表示方阵的大小, 4<=n<100
接下来是n行,每行有n个数据,可能是A,B,+,-中的某一个,中间用空格分开。
A,B都只出现一次。

要求输出一个整数,表示坦克从A区到B区的最少移动步数。
如果没有方案,则输出-1

例如:
用户输入:
5
A + - + -
- + - - +
- + + + -
+ - + - +
B + - + -

则程序应该输出:
10

资源约定:
峰值内存消耗 < 512M
CPU消耗 < 1000ms

思路:一开始感觉用dfs会超时,就用的bfs搜的,结果超内存了没过T.T(sad…)
dfs的话就判断一下和上一步走的符号相不相同就行了,相同的话直接跳过

#include<iostream>
#include<queue>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
int n,sx,sy,tx,ty,ans,m;
char a[105][105];
int nex[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
int vis[105][105]; 
map<char,int> mp;
void dfs(int x,int y,int num,int flag)
{
	//cout<<flag<<" ";
	if(x==tx && y==ty)
	{
		if(num<ans)
			ans=num;
		return;
	}
	for(int i=0;i<4;i++)
	{
		int xx=x+nex[i][0];
		int yy=y+nex[i][1];
		if(xx<0 || xx>=n || yy<0 || yy>=m || mp[a[xx][yy]]==flag || vis[xx][yy])
			continue;
		vis[xx][yy]=1;
		dfs(xx,yy,num+1,mp[a[xx][yy]]);
		vis[xx][yy]=0;
	}
}
int main()
{
	mp['+']=1;mp['-']=-1;
	scanf("%d",&n);
	string s;
	getchar();
	for(int i=0;i<n;i++)
	{
		getline(cin,s);
		m=0;
		for(int j=0;j<s.size();j++)
		{
			if(s[j]=='A'||s[j]=='B' || s[j]=='+'||s[j]=='-')
				a[i][m++]=s[j];
			if(a[i][m-1]=='A')
			{
				sx=i;sy=m-1;
			}
			if(a[i][m-1]=='B')
			{
				tx=i;ty=m-1;
			}
		}
	}
	ans=inf;
	vis[sx][sy]=1;
	dfs(sx,sy,0,0);
	if(!ans)
		cout<<-1<<endl;
	else
		cout<<ans<<endl;
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43693379/article/details/89441345