牛客网的题目如下:
题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。
如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bcced"的路径,
但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。
解决思想:
- 首先找到访问的起点,就是目标字符串的第一个字符与矩阵中的该字符的所有位置找出来。
- 使用bfs,将起点位置,上下左右,都去访问,每访问矩阵中一个字符的时候,会判断该字符是否合法,合法性的判断标准为:
字符所在的位置,行x,列y,属于矩阵中,该字符没有已经被访问过, - 如果该字符和目标字符串的当前需要访问的字符相同,说明当前字符有可能是路径中的一员,需要保存当前访问的信息,并放入队列,当访问它的邻居的时候,
就可以把当前访问的下标,拿来使用, - 只要访问矩阵节点的保存的下标,与字符串的最后一个字符的下标相同,说明目前字符串中的字符完全访问完毕,路径就表示找到。
实现的代码
本文实现的代码,只能满足牛客网的测试用例,当测试用例较全的时候,该代码无法通过,但可以用于学习下bfs,
#include<iostream>
#include<queue>
#include<cstring>
#include<cstdlib>
#include<cstdio>
using namespace std;
struct MyPoint {
int x;
int y;
int index;
MyPoint(int x1, int y1, int k):x(x1), y(y1), index(k) {
}};
class Solution {
public:
int row;
int col;
char *ma;
char *s;
bool hasPath(char *matrix, int rows, int cols, char *str) {
if ((*str) == ' ')
return false;
row = rows;
col = cols;
ma = matrix;
s = str;
vector < int >xs;
vector < int >ys;
int k;
int sLen = strlen(str);
int *visits = new int[rows * cols];
for (int i = 0; i < rows; i++)
for (int j = 0; j < cols; j++) {
k = i * cols + j;
if (matrix[k] == str[0]) {
xs.push_back(i);
ys.push_back(j);
}
} int xLen = xs.size();
for (int i = 0; i < xLen; i++) {
memset(visits, 0, sizeof(int) * rows * cols);
visits[xs[i] * col + ys[i]] = 1;
if (bfs(visits, xs[i], ys[i], 0, sLen))
return true;
}
return false;
}
bool bfs(int *visits, int startX, int startY, int curIndex, int sLen) {
static int dx[] = { 0, 1, 0, -1 };
static int dy[] = { -1, 0, 1, 0 };
queue < MyPoint * >q;
MyPoint *p0 = new MyPoint(startX, startY, curIndex);
q.push(p0);
while (!q.empty()) {
MyPoint *p1 = q.front();
q.pop();
printf("x=%d y=%d index=%d
", p1->x, p1->y, p1->index);
if (p1->index == (sLen - 1))
return true;
for (int i = 0; i < 4; i++) {
int xx = dx[i] + p1->x;
int yy = dy[i] + p1->y;
int kk = xx * col + yy;
if (visitIsLegal(xx, yy, visits)
&& ma[kk] == s[p1->index + 1]) {
MyPoint *p2 = new MyPoint(xx, yy, p1->index + 1);
visits[kk] = 1;
q.push(p2);
}
}
}
return false;
}
bool visitIsLegal(int x, int y, int *visits) {
if (x < 0 || x >= row)
return false;
if (y < 0 || y >= col)
return false;
int k = x * col + y;
if (visits[k])
return false;
return true;
}
};
int main()
{
char *matrix = "AAAAAAAAAAAA";
// char *str="AAAAAAAAAAAA";
char *str = "AAAAAAAAA";
int row = 3, col = 4;
Solution s;
cout << "res = " << s.hasPath(matrix, row, col, str) << endl;
}
dfs解法
bfs算法不适合,是因为把相邻的节点,都访问了,如果,目标串有很多相同的字符,会造成,其中的一些字符以后不会再访问了,
从而目标串中字符未访问完毕,所以本题目的正确解法,用深度优先搜索,每访问完,目标字符串中的字符,就需要访问它的下一字符,
对应为矩阵中,该字符的四周,当找到路径时,就不需要再去查找了,要从下往上返回true,就可以了,这个要注意哈
dfs 的代码如下
#include<iostream>
#include<queue>
#include<cstring>
#include<cstdlib>
#include<cstdio>
using namespace std;
/*
struct MyPoint {
int x;
int y;
int index;
MyPoint(int x1, int y1, int k):x(x1), y(y1), index(k) {
}};
*/
class Solution {
public:
int row;
int col;
char *ma;
char *s;
bool hasPath(char *matrix, int rows, int cols, char *str) {
if ((*str) == ' ')
return false;
row = rows;
col = cols;
ma = matrix;
s = str;
vector < int >xs;
vector < int >ys;
int k;
int sLen = strlen(str);
int *visits = new int[rows * cols];
for (int i = 0; i < rows; i++)
for (int j = 0; j < cols; j++) {
k = i * cols + j;
if (matrix[k] == str[0]) {
xs.push_back(i);
ys.push_back(j);
}
}
int xLen = xs.size();
for (int i = 0; i < xLen; i++) {
memset(visits, 0, sizeof(int) * rows * cols);
visits[xs[i] * col + ys[i]] = 1;
// cout<<"xs[i]"<<xs[i]<<" ys[i] "<<ys[i]<<endl;
if (dfs(visits, xs[i], ys[i], 0, sLen))
return true;
}
return false;
}
bool dfs(int *visits, int startX, int startY, int curIndex, int sLen) {
static int dx[] = { 0, 1, 0, -1 };
static int dy[] = { -1, 0, 1, 0 };
if(curIndex >= (sLen-1) )
return true;
// cout<<"startX ="<<startX<<" startY ="<<startY<<" curIndex = "<<curIndex<<endl;
for (int i = 0; i < 4; i++) {
int xx = dx[i] + startX;
int yy = dy[i] + startY;
int kk = xx * col + yy;
/*
if(startX == 0 && startY ==2){
cout<<"visits[3] ="<<visits[3]<<endl;
cout<<"xx= "<<xx<<" yy="<<yy<<endl;
cout<<"ma[kk]= "<<ma[kk]<<" s="<<s[curIndex+1]<<endl;
cout<<"visit index="<<visitIsLegal(xx, yy, visits)<<endl;
}
*/
if (visitIsLegal(xx, yy, visits)
&& ma[kk] == s[curIndex + 1]) {
visits[kk] = 1;
/*
if(xx == 0 && yy==3 && startX == 0 && startY ==2){
cout<<"ok"<<endl;
cout<<"index+1"<<curIndex+1<<endl;
}
*/
bool flag = dfs(visits,xx,yy,curIndex+1,sLen);
if( flag == true)
return true;
visits[kk] = 0;
}
}
return false;
}
bool visitIsLegal(int x, int y, int *visits) {
if (x < 0 || x >= row)
return false;
if (y < 0 || y >= col)
return false;
int k = x * col + y;
if (visits[k])
return false;
return true;
}
};
int main()
{
char *matrix = "AAAAAAAAAAAA";
// char *str="AAAAAAAAAAAA";
char *str = "AAAAAAAAA";
int row = 3, col = 4;
Solution s;
cout << "res = " << s.hasPath(matrix, row, col, str) << endl;
}