一、题目
二、解法
- 按照深度优先搜索DFS的解法:
1)确定初始节点:按照题目中来说,其实初始节点就是(0, 0),从(0, 0)出发看能到达哪个节点,能到达就+1
2)确定边界条件:由于DFS还是属于递归,因此需要有边界条件:当满足这个边界条件时,就应该到达了最终的节点不能继续往下走了;当不满足边界条件时,说明还可以继续往下遍历节点,同时返回的结果应该+1。因此边界条件应该是:节点到了网格外+行坐标和列坐标的数位之和大于k+该节点已经被遍历过,即i < 0 or i >= m or j < 0 or j >= n or (i//10 + i%10 + j//10 + j%10) > k or (i,j) in visited
3)在visited中记录该节点,表明它已经被遍历过(由于不需要回溯,因此遍历结束不需要再恢复到未被遍历的状态)
4)沿不同方向遍历下一个节点(调用dfs),同时+1
class Solution:
def dfs(self, i, j, m, n, k, visited):
if (i < 0 or i >= m or j < 0 or j >= n or (i//10 + i%10 + j//10 + j%10) > k or (i,j) in visited):
return 0
visited.add((i, j))
return self.dfs(i + 1, j, m, n, k, visited) + self.dfs(i - 1, j, m, n, k, visited) + self.dfs(i, j + 1, m, n, k, visited) + self.dfs(i, j - 1, m, n, k, visited) + 1
def movingCount(self, m: int, n: int, k: int) -> int:
visited = set()
return self.dfs(0, 0, m, n, k, visited)
- 但是,我最初的解法更复杂,没有考虑一次遍历即可:(x, y)是初始节点,(i, j)是目标节点,我是去遍历目标节点,看能不能到达。最后提交以后发现超时,确实这种更复杂,其实只需要遍历一次,能到达就+1即可。
class Solution {
public:
vector<int> dx = {0, 0, 1, -1};
vector<int> dy = {1, -1, 0, 0};
bool dfs(int x, int y, int i, int j, int m, int n, int k, vector<vector<bool>>& checked){
if (x == i && y == j){
return true;
}
checked[x][y] = true;
bool res = false;
for (int p = 0; p < 4; p++){
int a = x + dx[p];
int b = y + dy[p];
if (a >=0 && a < m && b >=0 && b < n && (a/10 + a%10 + b/10 + b%10) <= k){
if (checked[a][b] == false){
bool flag = dfs(a, b, i, j, m, n, k, checked);
if (flag){
res = true;
break;
}
}
}
}
checked[x][y] = false;
return res;
}
int movingCount(int m, int n, int k) {
vector<vector<bool>> checked(m, vector<bool>(n));
int res = 0;
for (int i = 0; i < m; i++){
for (int j = 0; j < n; j++){
bool flag = dfs(0, 0, i, j, m, n, k, checked);
if (flag){
res += 1;
}
}
}
return res;
}
};