[ Leetcode ] 面试题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。请问该机器人能够到达多少个格子?

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

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

题解:
刚开始的时候脑子抽了,想着这题直接遍历整个数组,再求每个位置的和不就好了...
然后提交结果给我灵魂一击,并且我还是没想反应过来!
想了一会儿,没想通,就看了眼答案,原来是这样:
比如说k = 5的时候,
(0, 0) ---> (0, 9) 是没有路径的。
结果呢,如果单纯的遍历整个数组,(0, 10)又变成可达的了,这显然不正确呀!
所以这题还是要用DFS,返回最后的maxNum。
最后的结果一定是所有可遍历的visit都置成了1。

class Solution {
public:
    int visit[101][101]  = {0}; 
    int maxNum = 1;
    int movingCount(int m, int n, int k) {
        visit[0][0] = 1;
        DFS(0, 0, m, n, k);
        return maxNum;
    }

    void DFS(int x, int y, int m, int n, int t) {
        int a[4] = {-1, 0, 1, 0};
        int b[4] = {0, 1, 0, -1};
        for(int k = 0; k < 4; k ++ ) {
            int u = a[k] + x, v = b[k] + y;
            if(u>=0 && u < m && v >=0 && v < n && visit[u][v] == 0) {
                int sum = CalculateSum(u) + CalculateSum(v);
                if(sum <= t) {
                    //cout << u << " " << v << ", sum:" << sum << endl;
                    visit[u][v] = 1;
                    maxNum += 1;
                    DFS(u, v, m, n, t);
                }
            }
        }
    }

    int CalculateSum(int num) {
        int sum = 0;
        while(num) {
            int a = num%10;
            sum += a;
            num = (num - a) / 10;
        }
        return sum;
    }
};

不过话说有点不会写DFS了.. hin生疏...
虽然看了题解,那也只是纠正了思路嘛
所以今天依旧是双百的一天呐!

上一篇:Acwing198 反素数


下一篇:LeetCode 769. 最多能完成排序的块