Leetcode: word search

思路:

简单搜索

总结:

1. 第一次做的时候忘记加 visited 数组了

2. 做完 kedebug 的搜索题, 简单搜索问题都变得像小绵羊一样温顺了

3. 还是 dfs 框架. 书上所写的 dfs 框架, 一般由 DFS, dfs 两个函数拼成, DFS 负责接收参数和变量初始化, dfs 负责一般情况下的遍历. 两个函数连用比仅用一个 dfs 要好的多, 因为减少了很多判断语句. 下面的代码, 相当于 DFS+ dfs.

代码:

#include <string>
#include <vector>
#include <iostream>
using namespace std;
const int MAXN = 200;
bool visited[MAXN][MAXN];
int dire[4][2] = {-1,0, 1,0, 0,-1, 0,1};
class Solution {
public:
vector<vector<char> > vec;
string str;
bool ans;

void dfs(const int &i, const int &j, const int &depth) {
if(depth == str.size()) {
ans = true;
return;
}
for(int d = 0; d < 4 && !ans; d ++) {// directions
int newi = i+dire[d][0], newj = j+dire[d][1];
if(newi >= 0 && newi <vec.size() && newj >= 0 && newj < vec[newi].size() && !visited[newi][newj]) {
if(vec[newi][newj] == str[depth]) {
visited[newi][newj] = 1;
dfs(newi, newj, depth+1);
visited[newi][newj] = 0;
}
}
}
}
bool exist(vector<vector<char> > &board, string word) {
vec = board;
str = word;
ans = false;
memset(visited, 0, sizeof(visited));
for(int i = 0; i < board.size()&&!ans; i ++) {
for(int j = 0; j < board[i].size()&&!ans; j++) {
if(board[i][j] == word[0]) {
visited[i][j] = true;
dfs(i, j, 1);
visited[i][j] = false;
}
}
}
return ans;
}
};