leetcode 剑指offer13 机器人运动范围

题干

描述

地上有一个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 <= n,m <= 100
• 0 <= k <= 20

示例

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

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

思路

最容易想到的肯定是DFS, 原来还想1<n,m<100的数据规模可能会超时, 但是最后的效果异常的好…

各数位之和

首先需要解决这个问题, 很容易想到利用%10和/10来分离各个数位, 逐个求和即可.

int eachSum(int x){
    int s = 0;
    while(x != 0) {
        s += x % 10;
        x = x / 10;
    }
    return s; 
}

dfs深搜遍历图

说到dfs首先需要解决的应该是递归边界, 这里很明显是当目标坐标越界的时候就必须返回了.

void dfs(int i, int j, int k){	//  dfs(i, j, k)表示可以走i, j这个坐标
    if(i<0 || i>=m || j<0 || j>=n)  return;		// 递归边界
    // 由于上面已经进行了坐标的合法判断, 因此能够执行到这, 说明i,j肯定是合法的.
    visit[i][j] = 1;	// 状态位
    cnt++;		// 记录可以走的坐标的数量
    
    /*
     *虽然保证了(i,j)一定合法, 但是上下左右的四个坐标不一定合法, 因此需要对他们做判断
     */
    
    // 向上搜索, 别忘了状态位的判断, 否则会导致死循环
    if(i>0 && (eachSum(i-1)+eachSum(j))<=k && visit[i-1][j]==0){
        dfs(i-1, j, k);
    }
    // 向下
    if(i+1<m && (eachSum(i+1)+eachSum(j))<=k && visit[i+1][j]==0){
        dfs(i+1, j, k);
    }
    // 向左
    if(j>0 && (eachSum(i)+eachSum(j-1))<=k && visit[i][j-1]==0){
        dfs(i, j-1, k);
    }
    // 向右
    if(j+1<n && (eachSum(i)+eachSum(j+1))<=k && visit[i][j+1]==0){
        dfs(i, j+1, k);
    }
    
    // 注意这题不需要下面这一句, 因为本题要求的是“所有满足条件的坐标”, 而不是“某一条特定的路径”
    // visit[i][j] = 0;
}

完整代码

class Solution {
public:
    int cnt = 0;
    static const int max = 101;
    int n, m;
    int visit[max][max];
    int eachSum(int x){
        int s = 0;
        while(x != 0) {
            s += x % 10;
            x = x / 10;
        }
        return s;
    }
    void dfs(int i, int j, int k){
        if(i<0 || i>=m || j<0 || j>=n)  return;
        visit[i][j] = 1;
        cnt++;
        if(i>0 && (eachSum(i-1)+eachSum(j))<=k && visit[i-1][j]==0){
            dfs(i-1, j, k);
        }
        if(i+1<m && (eachSum(i+1)+eachSum(j))<=k && visit[i+1][j]==0){
            dfs(i+1, j, k);
        }
        if(j>0 && (eachSum(i)+eachSum(j-1))<=k && visit[i][j-1]==0){
            dfs(i, j-1, k);
        }
        if(j+1<n && (eachSum(i)+eachSum(j+1))<=k && visit[i][j+1]==0){
            dfs(i, j+1, k);
        }
    }
    int movingCount(int mm, int nn, int k) {
        m = mm;
        n = nn;
        cnt = 0;
        memset(visit, 0, sizeof(visit));
        dfs(0, 0, k);
        return cnt;
    }
};

提交结果