[Leetcode/c++] 79. 单词搜索

题目

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

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

示例:

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

代码

#include <bits/stdc++.h>
#include <unordered_map>
#include <unordered_set>
using namespace std;
#define rep(i,a,n) for(int i=a; i<n; i++)
#define per(i,a,n) for(int i=n-1; i>=a; i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(), (x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
#define fr(x) freopen(x,"r",stdin)
#define fw(x) freopen(x,"w",stdout)
typedef vector<int> VI;
typedef vector<char> VC;
typedef vector<VI> VVI;
typedef vector<VC> VVC;
typedef long long ll;
typedef pair<int, int> PII;
typedef unordered_map<int, int> MII;
typedef double db;
ll gcd(ll a, ll b){return b?gcd(a%b, b):a;}
const ll mod = 1000000007;
// head 
class Solution {
public:
    int m, n;
    VVC grid;
    string pattern;
    bool vis[205][205];
    vector<PII> dir{mp(0, 1), mp(0, -1), mp(1, 0), mp(-1, 0)};
    bool exist(vector<vector<char>>& board, string word) {
        PII nxt;
        n = SZ(board); 
        m = SZ(board[0]); 
        rep(i, 0, n) rep(j, 0, m) vis[i][j] = false;
        grid = board;
        pattern = word;
        rep(i, 0, n) rep(j, 0, m) if(board[i][j] == word[0]){
            nxt = mp(i, j);
            vis[i][j] = true;
            if (dfs(nxt, 1)) return true;
            vis[i][j] = false;
        }
        return false;
    }
    bool valid(PII p){ return 0 <= p.fi && p.fi < n && 0 <= p.se && p.se < m;}
    bool dfs(PII point,int pos) {
        PII nxt;
        if(pos >= SZ(pattern)) return true;
        rep(i, 0, SZ(dir)) {
            nxt = mp(point.fi + dir[i].fi, point.se + dir[i].se);
            if(valid(nxt) && grid[nxt.fi][nxt.se] == pattern[pos] && !vis[nxt.fi][nxt.se]){
                vis[nxt.fi][nxt.se] = true;
                if(dfs(nxt, pos+1)) return true;
                vis[nxt.fi][nxt.se] = false; 
            }
        }
        return false;
    }
};

思路

1. 标准的网格搜索，DFS or BFS。
2. 可以开数组来记录访问状态， 同时，要注意全局变量的重置。