机器人的运动范围

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

回溯法:

public class Solution {
    public int movingCount(int threshold, int rows, int cols) {
        if (threshold == 0)
            return 1;
        if (threshold < 0)
            return 0;
        int[][] flag = new int[rows][cols];
        return move(threshold, rows, cols, flag, 0, 0);
    }
    
    public int move(int threshold, int rows, int cols, int[][] flag, int i, int j) {
        if (i >= rows || i < 0 || j >= cols || j < 0 || flag[i][j]==1 || bitSum(i)+bitSum(j)>threshold) {
            return 0;
        }
        flag[i][j] = 1;
        return move(threshold, rows, cols, flag, i+1, j)
            + move(threshold, rows, cols, flag, i-1, j)
            + move(threshold, rows, cols, flag, i, j+1)
            + move(threshold, rows, cols, flag, i, j-1)
            + 1;
    }
    
    public int bitSum(int n) {
        if (n == 0)
            return 0;
        int sum = 0;
        while (n != 0) {
            sum += n % 10;
            n /= 10;
        }
        return sum;
    }
}

 

posted @ 2016-10-05 16:58  Pickle  阅读(958)  评论(0编辑  收藏  举报