leetcode——面试题13.机器人的运动范围（C++）

题目描述：

  地上有一个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

题解：

基本条件：满足数位关系；
额外条件：是否可以进入。

1. 构造一个辅助二维数组（大小为m*n）；
2. 给第一行和第一列赋予值，满足条件赋予值1，不满足条件有两种情况：一种是基本条件不满足，赋予值-1，另一种是基本条件满足但是不能够进入，赋予值0；
3. 然后从下标[1,1]一直遍历到[m-1,n-1]得到所有满足条件的个数将其返回。

if(temp <= k && (matrix[i-1][j] == 1 || matrix[i][j-1] == 1)){
	matrix[i][j] = 1;
	cnt++;
}
else if(temp > k) matrix[i][j] = -1;
else matrix[i][j] = 0;

上图中Y 代表 1，N 代表 -1，X 代表 0

代码如下：

#include<iostream>
using namespace std;

int cal_number(int x){  //计算各位之和
    int temp = 0;
    do{
        temp += x % 10;
        x /= 10;
    }while(x);
    return temp;
}

int movingCount(int m, int n, int k){
    if(k == 0) return 1;
    int **matrix = new int*[m];
    for(int i = 0; i < m; i++) matrix[i] = new int[n];  //动态构造二维数组
    matrix[0][0] = 1;
    int cnt = 1;
    int temp;   //存放临时各位数之和
    for(int i = 1; i < n; i++){     //给第一行赋值
        temp = cal_number(i);
        if(temp <= k && matrix[0][i-1] == 1){
            matrix[0][i] = 1;
            cnt++;
        }
        else if(temp > k) matrix[0][i] = -1;
        else matrix[0][i] = 0;        
    }
    for(int i = 1; i < m; i++){     //给第一列赋值
        temp = cal_number(i);
        if(temp <= k && matrix[i-1][0] == 1){
            matrix[i][0] = 1;
            cnt++;
        }
        else if(temp > k) matrix[i][0] = -1;   
        else matrix[i][0] = 0;     
    }
    for(int i = 1; i < m; i++){     //从[1,1]遍历到[m-1,n-1]
        for(int j = 1; j < n; j++){
            temp = cal_number(i) + cal_number(j);
            if(temp <= k && (matrix[i-1][j] == 1 || matrix[i][j-1] == 1)){
                matrix[i][j] = 1;
                cnt++;
            }
            else if(temp > k) matrix[i][j] = -1;
            else matrix[i][j] = 0;
        }
    }
    return cnt;
}

int main(){
    int m, n, k;
    cin>>m>>n>>k;
    int move_cnt = movingCount(m, n, k);
    cout<<move_cnt<<endl;
    return 0;
}

复杂度分析：

 时间复杂度：O（M×N）
 空间复杂度：O（M×N）

题目来自leetcode，点击题目描述即可跳转到题目页面