题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 :
[a s a b;f d c c;e e s e]
矩阵中包含一条字符串"bcced"的路径,但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
思路
其实还是占格子然后内部循环的问题,但我觉得这个比之前那个机器人运动范围要难,因为每次循环时,先前试图走【上下左右】之前,该格子(坐标)已经标记上1了,在部分字符串符合要求,后面还在探究的时候,从其他方向跑到这个格子的可能性就被打断。这就是倒数第二行那个要设flag[index] = 0的原因(总之它必须得有,不然我python那个样例就是错误的输出)
python
# -*- coding:utf-8 -*-
class Solution:
def hasPath(self, matrix, rows, cols, path):
flag = [0 for _ in range(rows * cols)]
for i in range(0, rows): #这个范围是0到rows-1,我开始写成了rows-1所以总跳出去
for j in range(0, cols):
if self.helper(matrix, rows, cols, path, i, j, 0, flag):
return 1
return 0
def helper(self, matrix, rows, cols, path, i, j, k, flag):
index = j + cols * i
if i >= rows or i < 0 or j >= cols or j < 0 or matrix[index] != path[k] or flag[index] == 1:
return 0
if k >= len(path) - 1:
return 1
flag[index] = 1
if(
self.helper(matrix, rows, cols, path, i - 1, j, k + 1, flag) or\
self.helper(matrix, rows, cols, path, i + 1, j, k + 1, flag) or\
self.helper(matrix, rows, cols, path, i, j - 1, k + 1, flag) or\
self.helper(matrix, rows, cols, path, i, j + 1, k + 1, flag)):
return 1
flag[index] = 0
# 这里我一开始以为是多余的,如果不设回0,当path很长的时候,就不能从其他路径绕回来,因为部分字符符合要求却已经在循环内把这个格设为了1。
return 0
if __name__ =='__main__':
a=Solution()
print(a.hasPath("ABCEHJIGSFCSLOPQADEEMNOEADIDEJFMVCEIFGGS",5,8,"SGGFIECVAASABCEHJIGQEM"))
C++
#include <iostream>
using namespace std;
//这里的char[] 是生成与matrix规格相同的空数组,其实就是占个与matrix一毛一样的位置
class Solution {
public:
bool hasPath(char* matrix, int rows, int cols, char* str)
//逐个检索的函数
{
int len = strlen(matrix);
int flag[len]={0};
for(int i = 0; i < rows; i ++){
for(int j = 0; j < cols; j ++){
if(helper(matrix, rows, cols, i, j, str, 0, flag)){
return true;
}
}
}
return false;
};
bool helper(char* matrix, int rows, int cols, int i, int j, char* str, int k, int* flag)
// matrix:矩阵 row、cols:矩阵的行列数 i,j检索矩阵时的坐标 str:所要找的字符串 k:第k个字符串,从0开始找 flag:记录占格子用的序列,大小与matrix相等
{
int index = i * cols + j;
if(i < 0 || i >= rows || j < 0 || j >= cols || matrix[index] != str[k] || flag[index] == 1){
//下标不符合,index对应的值不为和字符数组中的不一致,或者该index已经被访问,这些情况只要有符合的就返回false
//只有上面的所有情况都不符合,也就是值相等,且没有访问过,下标不符合
return false;
};
if(k == strlen(str) - 1){
return true;
};
flag[index] = 1;
if(helper(matrix, rows, cols, i - 1, j, str, k + 1, flag)
||helper(matrix, rows, cols, i + 1, j, str, k + 1, flag)
||helper(matrix, rows, cols, i, j - 1, str, k + 1, flag)
||helper(matrix, rows, cols, i , j + 1, str, k + 1, flag)){
return true;
};
flag[index] = 0;
return false;
};
};
int main(){
bool hasPath();
Solution a;
cout<<a.hasPath("abcehfvjs",3,3,"aev")<<endl;
};
后话
这个代码我搞了好久,首先是python的从0开始计数的方式还不太熟练;然后c++的指针也忘得差不多了。
我看到牛客有大佬的代码运算速度是我的3倍,好像用的是向量,哎,等我之后在对比研究吧。
参考
C++代码参考blog:https://blog.csdn.net/qq_28081081/article/details/80891240