机器人的运动范围

2019独角兽企业重金招聘Python工程师标准>>>

题目描述

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

【解决】

① dfs

public class Solution {
    public static void main(String[] args){
        System.out.println(movingCount(5, 10, 10) + "[21]");
        System.out.println(movingCount(15, 20, 20) + "[359]");
        System.out.println(movingCount(12, 1, 1) + "[1]");
        System.out.println(movingCount(-10, 10, 10) + "[0]");
    }
    public static int movingCount(int threshold, int rows, int cols) {
        boolean[] isvisited = new boolean[rows * cols];
        int count = dfs(threshold,rows,cols,0,0,isvisited);
        return count;
    }
    public static int dfs(int threshold,int rows,int cols,int i,int j,boolean[] isvisited){
        int count = 0;
        if (canMove(threshold,rows,cols,i,j,isvisited)){
            isvisited[i * cols + j] = true;
            count = 1 + dfs(threshold,rows,cols,i - 1,j,isvisited)
                    + dfs(threshold,rows,cols,i + 1,j,isvisited)
                    + dfs(threshold,rows,cols,i,j - 1,isvisited)
                    + dfs(threshold,rows,cols,i,j + 1,isvisited);
        }
        return count;
    }
    public static boolean canMove(int threshold,int rows,int cols,int i,int j,boolean[] isvisited){
        if (i >= 0 && i < rows &&  j >= 0 && j < cols && getDigits(i) + getDigits(j) <= threshold
                 && ! isvisited[i * cols + j]){
            return true;
        }
        return false;
    }

    public static int getDigits(int num){
        int sum = 0;
        while(num != 0){
            sum += num % 10;
            num /= 10;
        }
        return sum;
    }
}

转载于:https://my.oschina.net/liyurong/blog/1784131