《剑指offer》机器人的运动范围(Java)_剑指offer机器人的运动范围-java代码测试用例-CSDN博客

本文介绍了一种计算机器人在特定条件下可到达格子数量的算法。机器人在一个m行n列的方格中移动，每步只能向左、右、上、下移动一格，且不能进入数位和超过k的格子。文章提供了两种实现方式，一种使用递归深度优先搜索，另一种使用简化版的深度优先搜索。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

AC代码

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<0||rows<0||cols<0)
            return 0;
 
        boolean[] visited = new boolean[rows*cols];
        for(int i = 0;i<rows*cols;i++){
            visited[i] = false;
        }
        int count = movintCountCore(threshold,rows,cols,0,0,visited);
 
        return count;
    }
     
    public int movintCountCore(int threshold,int rows,int cols,int row,int col,boolean[] visited){
        int count = 0;
        if(check(threshold,rows,cols,row,col,visited)){
            visited[row*cols+col] = true;
 
            count=1 + movintCountCore(threshold,rows,cols,row+1,col,visited)
                    + movintCountCore(threshold,rows,cols,row-1,col,visited)
                    + movintCountCore(threshold,rows,cols,row,col+1,visited)
                    + movintCountCore(threshold,rows,cols,row,col-1,visited);
        }
 
        return count;
    }
 
    boolean check(int threshold,int rows,int cols,int row,int col,boolean[] visited){
        if(row>=0&&col>=0&&row<rows&&col<cols
            &&getDigitSum(row)+getDigitSum(col)<=threshold
            &&!visited[row*cols+col])
            return true;
 
        return false;
    }
 
    int getDigitSum(int num){
        int sum = 0;
        while(num>0){
            sum += num%10;
            num /= 10;
        }
 
        return sum;
    }
     
}

二刷，简化版代码，思路还是一样的。

public class Solution {
    int[] dx={0,1,0,-1};int[] dy={1,0,-1,0};
    int res=1;
    public int movingCount(int threshold, int rows, int cols)
    {
        if(threshold<0||rows<=0||cols<=0)
            return 0;
        boolean [][] isState= new boolean [rows][cols];
        dfs(0,0,threshold,rows,cols,isState);
        return res;
    }

    void dfs(int x,int y,int k,int rols,int cols,boolean [][]isState){
        isState[x][y]=true;
        for(int i=0;i<4;i++){
            int a=x+dx[i],b=y+dy[i];
            int t=a%10+a/10+b%10+b/10;
            if(a>=0&&b>=0&&a<rols&&b<cols&&t<=k&&isState[a][b]==false){
                res+=1;
                dfs(a,b,k,rols,cols,isState);
            }
        }
    }
   
}