数据结构与算法-回溯法

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

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    bool hasPath(char* matrix, int rows, int cols, char* str)
    {
        if(NULL == matrix || NULL == str)
        {
            return false;
        }

        // 已进入的标志位，初始化为全false
        vector<bool> flags(rows*cols, false);

        //遍历二维数组，找到一个与str第一个元素匹配的矩阵元素，再递归判断四周是否符合条件
        for(int i = 0; i< rows; i++) {
            for(int j=0; j< cols; j++)
            {
                if(judge(matrix, rows, cols, i, j, flags, str, 0))
                {
                    return true;
                }
            }
        }

        return false;
    }

    bool judge(char* matrix, int rows, int cols, int i, int j, vector<bool> &flags, char* str, int k)
    {
        int index = i * cols + j;
        //找不到匹配元素，递归终止
        if(i < 0 || j < 0 || i>=rows || j>=cols || matrix[index] != str[k] || flags[index] == true)
        {
            return false;
        }

        //找到匹配元素，若已经到达str末尾了，说明全部都匹配成功了, 返回true
        if(str[k+1] == '\0')
            return true;

        //找到匹配元素，标记下
        flags[index] = true;

        // 前后左右继续往找匹配下一个元素的，找不到，还原(回溯，递归查找)
        if(judge(matrix, rows, cols, i-1, j, flags, str, k+1) ||
           judge(matrix, rows, cols, i, j-1, flags, str, k+1) ||
           judge(matrix, rows, cols, i+1, j, flags, str, k+1) ||
           judge(matrix, rows, cols, i, j+1, flags, str, k+1))
        {
            return true;
        }

        // 走到这里，说明这一条路不通，还原标志位，再试其他路径
        flags[index] = false;
        return false;
    }
};

int main() {
    int rows = 3;
    int cols = 4;
    char* matrix = "abcesfcsadee";
    char* str = "bcced";
    Solution s;
    cout << s.hasPath(matrix, rows, cols, str) << endl;
}