leetcode Word Search回溯算法

来源:互联网 发布:局域网ip冲突 软件 编辑:程序博客网 时间:2024/06/06 15:39

Given a 2D board and a word, find if the word exists in the grid.

The word can be constructed from letters of sequentially adjacent cell, where "adjacent" cells are those horizontally or vertically neighboring. The same letter cell may not be used more than once.

For example,
Given board =

[  ["ABCE"],  ["SFCS"],  ["ADEE"]]
word = "ABCCED", -> returns true,
word = "SEE", -> returns true,

word = "ABCB", -> returns false.

感觉这个题还是挺经典的,跟走迷宫差不多,每次可以走四个方向,上下左右,且走过的不能再走,关键是捋清思路,首先需要确定各自的入口点,通过遍历碰到相同的字符则进入,如果找到了就不继续遍历,然后是深度的确定,深度应该是字符串的长度,而不是所有的可能的排列,每走一步,深度加一,超过字符串长度则结束,寻路的时候分上下左右,先标记访问过,再进行回溯,然后标记为未访问,加入的条件是找到相同的字符再加入,而不是所有的可能都加入然后判断与word是否相等。

访问数组初始化的时候可以使用这句,不用占用更多的空间,比如申请200*200的数组。

visited.resize(row, vector<bool>(col));//初始化visited

下面就是代码

class Solution {public:string str;int row;int col;bool res = false;vector<vector<bool>> visited;//辅助数组,用于记录是否访问过vector<vector<char>> boardcopy;//先确定格子的的入口点,然后每走一步深度加一,如果深度超过字符串的长度则结束void backtracking(int r, int c, int dep){if (dep == str.size()){res=true;return;}if (r>0 && !res&&!visited[r - 1][c] && boardcopy[r - 1][c] == str[dep])//上:能往上走,res还没true,没被访问过,且与字符串相等{visited[r - 1][c] = true;backtracking(r-1, c, dep + 1);visited[r - 1][c] = false;}if (r<row-1 && !res&&!visited[r+1][c] && boardcopy[r+1][c] == str[dep])//下:能往下走,res还没true,没被访问过,且与字符串相等{visited[r+ 1][c] = true;backtracking(r + 1, c, dep + 1);visited[r + 1][c] = false;}if (c>0 && !res&&!visited[r][c - 1] && boardcopy[r][c - 1] == str[dep])//左:还没true,没被访问过,且与字符串相等{visited[r][c - 1] = true;backtracking(r, c-1, dep + 1);visited[r][c-1] = false;}if (c<col-1 && !res&&!visited[r][c + 1] && boardcopy[r][c + 1] == str[dep])//右:还没true,没被访问过,且与字符串相等{visited[r][c + 1] = true;backtracking(r, c+1, dep + 1);visited[r][c+ 1] = false;}}bool exist(vector<vector<char>>& board, string word) {row = board.size();if (!row) return false;col = board[0].size();str = word;boardcopy = board;visited.resize(row, vector<bool>(col));//初始化visitedfor (int i = 0; i < row&&!res; i++){for (int j = 0; j < col&&!res; j++){if (board[i][j]==word[0])//入口点{visited[i][j] = true;//入口点标记访问过    backtracking(i, j, 1);//dep为1visited[i][j] = false;//记得返回成未访问}}}return res;}};


0 0
原创粉丝点击