【剑指offer】Java实现-机器人的运动范围

题目描述

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

思路

这道题和【矩阵中的路径】的思路是差不多的，都是使用回溯法。
用一个数组visited[][]表示地上的方格是否已被访问过
从起点(0, 0)开始，沿着路径（上下左右）走，每到一个符合条件的方格，
数量就加1，并标识该方格已被访问过

代码

public class Solution {
    public int movingCount(int threshold, int rows, int cols){
        //如果方格行列小于0，或者k<0，则返回0
        if(rows<=0 || cols<=0 || threshold<0)
            return 0;
        //定义一个visited[][]数组，表示是否方格已经被访问过
        boolean[][] visited=new boolean[rows][cols];
        return movingCount(threshold,rows,cols,0,0,visited);
    }
    //判断方格是否被访问过，r 表示走过的行，c 表示走过的列
    private int movingCount(int threshold,int rows,int cols,int r,int c,boolean[][] visited){
        //如果超出方格范围，返回0
        if(r>=rows || r<0 || c>=cols || c<0)
            return 0;
        //如果方格未被访问或各位数字之和大于threshold,返回0
        if(visited[r][c] || isBiggerThanThreshold(r,c,threshold))
            return 0;
        //走过某一坐标，visited[r][c]改为true
        visited[r][c]=true;
        //能够到达的格子数=原点+向右移动+向左移动+向上移动+向下移动的格子数
        return 1+movingCount(threshold,rows,cols,r+1,c,visited)
            +movingCount(threshold,rows,cols,r-1,c,visited)
            +movingCount(threshold,rows,cols,r,c+1,visited)
            +movingCount(threshold,rows,cols,r,c-1,visited);
    }
    //判断各位数字之和是否大于threshold
    private boolean isBiggerThanThreshold(int r,int c,int threshold){
        int sum=0;
        while(r>0){
            sum+=r%10;
            r/=10;
        }
        while(c>0){
            sum+=c%10;
            c/=10;
        }
        return sum>threshold;
    }
}

目录