给定一个二维网格和一个单词,找出该单词是否存在于网格中。
单词必须按照字母顺序,通过相邻的单元格内的字母构成,其中“相邻”单元格是那些水平相邻或垂直相邻的单元格。同一个单元格内的字母不允许被重复使用。
示例:
board =
[
[‘A’,‘B’,‘C’,‘E’],
[‘S’,‘F’,‘C’,‘S’],
[‘A’,‘D’,‘E’,‘E’]
]
给定 word = “ABCCED”, 返回 true
给定 word = “SEE”, 返回 true
给定 word = “ABCB”, 返回 false
dfs+剪枝操作,分别向四个方向寻找即可。
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
static const int N = 200;
bool mark[N][N];
int dx[4] = {1,-1,0,0};
int dy[4] = {0,0,1,-1};
int n,m;
bool exist(vector<vector<char>>& board, string word) {
n = board.size();
m = board[0].size();
for (int i = 0; i < n; ++i)
{
for(int j = 0; j < m; ++j)
{
mark[i][j] = true;
if(board[i][j] == word[0] && dfs(board, word, i, j, 0)) return true;
mark[i][j] = false;
}
}
return false;
}
bool dfs(vector<vector<char>>& board, string& word, int x, int y, int index)
{
if(index == word.size() -1) return true;
for(int i = 0; i < 4; ++i)
{
int a = x + dx[i];
int b = y + dy[i];
if(a < 0 || a >= n || b < 0 || b >= m) continue;
if(mark[a][b] == true) continue;
if(board[a][b] != word[index + 1]) continue;
mark[a][b] = true;
if(dfs(board, word, a, b, index + 1)) return true;
mark[a][b] = false;
}
return false;
}
};