牛客网中矩阵中的路径

牛客网的题目如下:

题目描述
请设计一个函数,用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始,每一步可以在矩阵中向左,向右,向上,向下移动一个格子。
如果一条路径经过了矩阵中的某一个格子,则该路径不能再进入该格子。 例如 a b c e s f c s a d e e 矩阵中包含一条字符串"bcced"的路径,
但是矩阵中不包含"abcb"路径,因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后,路径不能再次进入该格子。

解决思想:

  1. 首先找到访问的起点,就是目标字符串的第一个字符与矩阵中的该字符的所有位置找出来。
  2. 使用bfs,将起点位置,上下左右,都去访问,每访问矩阵中一个字符的时候,会判断该字符是否合法,合法性的判断标准为:
    字符所在的位置,行x,列y,属于矩阵中,该字符没有已经被访问过,
  3. 如果该字符和目标字符串的当前需要访问的字符相同,说明当前字符有可能是路径中的一员,需要保存当前访问的信息,并放入队列,当访问它的邻居的时候,
    就可以把当前访问的下标,拿来使用,
  4. 只要访问矩阵节点的保存的下标,与字符串的最后一个字符的下标相同,说明目前字符串中的字符完全访问完毕,路径就表示找到。

实现的代码

本文实现的代码,只能满足牛客网的测试用例,当测试用例较全的时候,该代码无法通过,但可以用于学习下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) == '\0')
	    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\n", 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) == '\0')
	    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;
}

posted on 2019-10-24 15:00  盛夏落木  阅读(207)  评论(0编辑  收藏  举报

导航