剑指 Offer 13. 机器人的运动范围

题目描述

地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动，它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。例如，当k为18时，机器人能够进入方格 [35, 37] ，因为3+5+3+7=18。但它不能进入方格 [35, 38]，因为3+5+3+8=19。请问该机器人能够到达多少个格子？

示例 1：

输入：m = 2, n = 3, k = 1
输出：3

示例 2：

输入：m = 3, n = 1, k = 0
输出：1

提示：

• 1 <= n,m <= 100
• 0 <= k <= 20

---

        本题的最终目的是要知道有多少个点满足要求，那么首先想到的想法是遍历一遍这个二维数组，然后看每个位置的左边（i，j）是否满足数位和小于k的要求，然后统计满足要求的点的个数。但是这种方法存在问题就是题目要求每次只能移动一格，但我们遍历二维数组时会有从一行的末尾跳到另一行的行首的情况，所以只能通过深度优先搜索的方法来查找所有可能到达的点。

        本题的做法就是DFS，从（0,0）这个点出发，遍历它的上下左右四个点，如果满足条件那么久加入到结果集中，最终放回结果集的长度。代码如下：

class Solution(object):
    def movingCount(self, m, n, k):
        """
        :type m: int
        :type n: int
        :type k: int
        :rtype: int
        """
        def sum(a,b):
            res=0
            for i in str(a)+str(b):
                res+=int(i)
            return res>k
        def DFS(i,j):
            if i<0 or j<0 or i>m-1 or j>n-1 or sum(i,j) or(i,j) in s:
                return
            s.add((i,j))
            DFS(i+1,j)
            DFS(i-1,j)
            DFS(i,j+1)
            DFS(i,j-1)
        s=set()
        DFS(0,0)
        return len(s)

        在这里采用了set来当做结果集，因为set默认是不保存重复结点的，所以同时可以作为一个访问数组来判断每个点有没有被遍历过。

        因为每个点都会计算它右边和下边的点，所以说其实遍历左边和上边的点是无效的，这两个点一定是从右边或者下边递归过来的。所以可以直接省去遍历左边和上边这两个点的操作，这样还省时间。

class Solution(object):
    def movingCount(self, m, n, k):
        """
        :type m: int
        :type n: int
        :type k: int
        :rtype: int
        """
        def sum(a,b):
            res=0
            for i in str(a)+str(b):
                res+=int(i)
            return res>k
        def DFS(i,j):
            if i<0 or j<0 or i>m-1 or j>n-1 or sum(i,j) or(i,j) in s:
                return
            s.add((i,j))
            DFS(i+1,j)
            DFS(i,j+1)
        s=set()
        DFS(0,0)
        return len(s)