【蓝桥杯】试题 历届真题 穿越雷区【第六届】【决赛】【A组】

一道比较简单的dfs搜索题。本题的两个要点：一是A开始B结束，二是走的过程中+-交替。

具体的解题过程写在注释中了。

个人经验是把字符矩阵转换成数字矩阵在判断条件时会简单一些。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<math.h>
#include<set>
#include<string>
using namespace std;
int n;
int begini, beginj, endi, endj;//A,B的坐标
int ans = 99999;//记录最终的最小步数

//初始化方阵，+为1，-为-1，A或B为0；同时记录A,B的坐标
void init(vector<vector<int> >&grid)
{
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < n; j++)
		{
			char ch;
			cin >> ch;
			if (ch == '+')
			{
				grid[i][j] = 1;
			}
			else if (ch == '-')
			{
				grid[i][j] = -1;
			}
			else if (ch == 'A')
			{
				grid[i][j] = 0;
				begini = i;
				beginj = j;
			}
			else if (ch == 'B')
			{
				grid[i][j] = 0;
				endi = i;
				endj = j;
			}
		}
	}
}
//i，j为方阵坐标，len为走了多少步，pre记录上一步是+还是-，used记录当前位置有没有走过一次
void dfs(int i,int j,int len,int pre,vector<vector<int> >&used, vector<vector<int> >& grid)
{
	if (i == endi && j == endj)
	{
		ans = min(ans, len);
		return;
	}
	//当len>ans已经不可能是结果了，直接返回
	//经实测都可以通过。加上len>ans用时31ms，不加用时453ms
	if (i < 0 || j < 0 || i >= grid.size() || j >= grid[0].size()||used[i][j]==1||len>ans)
	{
		return;
	}
	//if判断是否符合+-交替，pre==0是开始情况
	if (grid[i][j]+pre==0||pre==0)
	{
		used[i][j] = 1;
        //进行四个方向的搜索
		dfs(i + 1, j, len + 1,grid[i][j], used, grid);
		dfs(i - 1, j, len + 1,grid[i][j], used, grid);
		dfs(i, j + 1, len + 1,grid[i][j], used, grid);
		dfs(i, j - 1, len + 1,grid[i][j], used, grid);
		used[i][j] = 0;
	}
	
	
	
}
int main()
{
	cin >> n;
	vector<vector<int> >grid(n,vector<int>(n));
	init(grid);
	vector<vector<int> >used(n, vector<int>(n, 0));
	dfs(begini, beginj, 0, 0, used,grid);
	ans = ans == 99999 ? -1 : ans;
	cout << ans;

	return 0;
}