【深度优先搜索】机器人的运动范围,中等难度,需要注意初始节点

一、题目

在这里插入图片描述

二、解法

  • 按照深度优先搜索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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值