LeetCode 79.单词搜索

博主 909 2020-09-13

给定一个二维网格和一个单词,找出该单词是否存在于网格中。

单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。

 

示例:

board =
[
['A','B','C','E'],
['S','F','C','S'],
['A','D','E','E']
]

给定 word = "ABCCED", 返回 true
给定 word = "SEE", 返回 true
给定 word = "ABCB", 返回 false
 

提示:

board 和 word 中只包含大写和小写英文字母。
1 <= board.length <= 200
1 <= board[i].length <= 200
1 <= word.length <= 10^3

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/word-search
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


实际上就是图的dfs遍历
每次访问一个新的位置,比对对应的字符是否存在,注意不能走回头路

代码:

class Solution {
private:
    bool visit[205][205] = {0};
    string word;
    vector<vector<char> > board;
    int inc[4][2]  = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
public:
    bool exist(vector<vector<char>>& board, string word) {
        this->word = word;
        this->board = board;
        char firstLetter = word[0];
        vector<pair<int, int> > f;
        for (int i = 0; i < board.size(); ++i) {
            for (int j = 0; j < board[i].size(); ++j) {
                if (board[i][j] == firstLetter) {
                    f.push_back(pair<int, int>(i, j));
                }
            }
        }

        for (int i = 0; i < f.size(); ++i) {
            if (dfs(f[i].first, f[i].second, 0)) {
                return true;
            }
        }
        return false;
    }
    
    bool dfs(int i, int j, int pos){
        if (pos == word.size()-1) {// 到达最后一个字符处 需要验证是否为该字母
            if (board[i][j] == word[pos])
                return true;
            else
                return false;
        }
        
        if (pos > word.size()) {
            return false;
        }
        
        if (board[i][j] == word[pos]) {
            for (int k = 0; k < 4; ++k) {
                if (canVisit(i, j, k)) {
                    visit[i][j] = true;
                    if (dfs(i + inc[k][0], j + inc[k][1], pos+1)) return true;
                    visit[i][j] = false;
                }
            }
        }

        return false;
    }
    
    // 判断i,j加上inc[k]的增量后是否可达
    bool canVisit(int i, int j, int k){
        int m = i + inc[k][0];
        int n = j + inc[k][1];
        // 越界检查
        if (m < 0 || n < 0) {
            return false;
        }
        if (m >= board.size() || n >= board[m].size()) {
            return false;
        }
        
        // 是否访问过了
        if (visit[m][n] == true) {
            return false;
        }
        return true;
    }
};