牛客_机器人的运动范围

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zhangvalue/article/details/88218815

题目描述

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

分析:

在考虑到机器人在移动的过程中都，是走上下左右四个方向，考虑使用回溯算法

1、从(0,0)开始移动，每成功走一步标记当前位置为的flag置为1,然后从当前位置往四个方向移动，

返回1 + 4 个方向的移动值之和。

2、移动前，判断当前节点是否可达的标准为：

1）当前节点在矩阵内；

2）当前节点未被访问过；

3）当前节点满足threshold限制。

public class Solution {
 public int movingCount(int threshold, int rows, int cols) {
     //设置flag作为标记，如果flag为1说明已经遍历访问过了
       int[][] flag = new int[rows][cols];
       return moving(threshold, rows, cols, flag, 0, 0);
    }
    public int moving(int threshold, int rows, int cols, int[][] flag, int i, int j){
        //判断是否是越界，或者超出k或者已经被访问过
        //因为坐标是从0开始计算所以i和j都不能等于rows和cols
        if(threshold <= 0 || i >= rows || i< 0 || j >= cols 
           || j < 0 || (flag[i][j] == 1) 
           ||(sum(i) + sum(j) > threshold)){
            return 0;
        }
        //到了当前点，记录一下当前点被访问
        flag[i][j] = 1;
        //回溯当前点的左右下上+当前点的情况
        //同时return作为递归循环结束的条件
        return moving(threshold, rows, cols, flag, i - 1, j)
            +moving(threshold, rows, cols, flag, i + 1, j)
            +moving(threshold, rows, cols, flag, i, j - 1)
            +moving(threshold, rows, cols, flag, i, j + 1)
            + 1;   
    }
    //将输入的一个数的
    public int sum(int i ){
        if(i == 0){return i ;}
        int sum = 0;
        while(i != 0){
            //先取到个位数
            sum += i % 10;
            //再取高位
            i /= 10;
        }
        return sum;
    }
}