【刷算法】求机器人的运动范围

题目描述

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

分析

对于每个到(x,y)的机器人，有四种移动可能，上下左右，即：(x+1,y)，(x-1,y)，(x,y-1)，(x,y+1)；

决定该位置是否合法要检查这么几个方面：

1. x、y坐标不能超出rows和cols的界限
2. x、y的坐标数位之和不能大于k

代码实现

function movingCount(k, rows, cols)
{
    var flags = [];
    for(var i = 0;i < rows;i++) {
        flags.push([]);
        for(var j = 0;j < cols;j++) {
            flags[i].push(0);
        }
    }
    
    return steps(0,0,rows,cols,flags, k);
}
function steps(x, y, rows, cols, flags, k){
    if(x <0 || x >= rows || y < 0 || y >= cols || flags[x][y] === 1 || (bitSum(x) + bitSum(y) > k) )
        return 0;
    flags[x][y] = 1;
    return steps(x-1, y, rows, cols, flags, k) + steps(x+1, y, rows, cols, flags, k) + steps(x, y-1, rows, cols, flags, k) + steps(x, y+1, rows, cols, flags, k) + 1;
}

function bitSum(n){
    var sum = 0;
    while(n >= 1){
        sum += n%10;
        n = Math.floor(n/10);

    }
    return sum;
}