请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则之后不能再次进入这个格子。 例如 下面 这样的3 X 4 矩阵中包含一条字符串”bfce”的路径,但是矩阵中不包含”abfb”路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
我们采用回溯法解决问题。
首先,在矩阵中任意选择一个格子作为路径起点。假设矩阵中某个格子的字符为ch,并且这个格子将对应与路径上的第i个字符。如果路径上的第i个字符不是ch,那么这个格子不可能处在路径上的第i个位置。如果路径上的第i个字符正好是ch,那么到相邻的格子寻找路径上的第i+1个字符。除矩阵边界上的格子之外,其他格子都有4个相邻格子。
重复这个过程,直到路径上的所有字符都在矩阵中找到相应位置。
/**
* 回溯搜索算法
*
* @param matrix 输入矩阵
* @param rows 矩阵行数
* @param cols 矩阵列数
* @param str 要搜索的字符串
* @param visited 访问标记数组
* @param x 当前处理的行号
* @param y 当前处理的列号
* @param pathLength 已经处理的str中字符个数
* @return 是否找到 true是,false否
*/
class Solution {
public:
bool hasPath(char* matrix, int rows, int cols, char* str)
{
if(matrix==NULL || rows<1 || cols<1 || str==NULL)
return false;
//标记有没有访问到该点。
bool *visited=new bool[rows*cols];
memset(visited,false,rows*cols);//初始化每个值为未访问
//记录已经搜索到满足条件的字符串的字符个数
//在当前条件不满足,需要回到上一层,
int pathLength=0;
for(int i=0;i<rows;++i)
for(int j=0;j<cols;++j)
if (hasPathCore(matrix, rows, cols, str, visited, i, j, pathLength))
return true;
delete[] visited;
return false;
}
bool hasPathCore(const char* matrix, int rows, int cols,
const char* str, bool* visited, int x, int y, int& pathLength)
{
//str字符串搜索完成
if(str[pathLength]=='\0')
return true;
bool hasPath = false;
// 判断位置是否合法
//因为是一维数组存放二维的值,row * cols + col值就是相当于二维数组的(row,col)在一维数组的下标
if(x < rows && x >= 0 && y >= 0 && y < cols &&
matrix[x * cols + y] == str[pathLength] &&
!visited[x * cols + y]){
visited[x * cols + y] = true;//标记已经访问
++pathLength;//已经处理的str中字符个数+1
// 按左上右下进行回溯
hasPath = hasPathCore(matrix, rows, cols, str, visited, x, y - 1, pathLength)
|| hasPathCore(matrix, rows, cols, str, visited, x - 1, y, pathLength)
|| hasPathCore(matrix, rows, cols, str, visited, x, y + 1, pathLength)
|| hasPathCore(matrix, rows, cols, str, visited, x + 1, y, pathLength);
if (!hasPath) {//不满足
--pathLength;//返回上一步
visited[x * cols + y] = false;
}
}
return hasPath;
}
};