剑指offer:66.机器人的运动范围

最新推荐文章于 2021-09-20 14:11:31 发布

原创最新推荐文章于 2021-09-20 14:11:31 发布 · 154 阅读

0 ·

CC 4.0 BY-SA版权

Java 同时被 3 个专栏收录

183 篇文章

订阅专栏

数据结构

154 篇文章

订阅专栏

剑指Offer提高班

67 篇文章

订阅专栏

本文介绍了一种基于深度优先搜索的算法，用于计算机器人在特定条件下可以到达的格子数量。机器人在一个m行n列的网格中移动，避免进入数位和超过k的格子。通过初始化每个格子的数位和并使用DFS遍历，算法有效地计算了可达格子的总数。

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

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

//思路：深度优先遍历
public class Solution {
    
    int[][]next={{0,-1},{0,1},{-1,0},{1,0}};
    int count=0,rows,cols,threshold;
    int[][]digitSum;
    
    public int movingCount(int threshold, int rows, int cols)
    {
        this.rows=rows;
        this.cols=cols;
        this.threshold=threshold;
        initSumOne();
        boolean[][]marked=new boolean[rows][cols];
        dfs(marked,0,0);
        return count;
    }
    
    public void dfs(boolean[][]marked,int r,int c){
        if(c>=cols||c<0||r>=rows||r<0||marked[r][c])
            return;
        marked[r][c]=true;
        if(digitSum[r][c]>threshold)return;
        count++;
        for(int[]n:next)
            dfs(marked,r+n[0],c+n[1]);
    }
    
    public void initSumOne(){
        int[]digitSumOne=new int[Math.max(rows,cols)];
        for(int i=0;i<digitSumOne.length;i++){
            int n=i;
            while(n>0){
                digitSumOne[i]+=n%10;
                n/=10;
            }
        }
        digitSum=new int[rows][cols];
        for(int i=0;i<rows;i++)
            for(int j=0;j<cols;j++)
                digitSum[i][j]=digitSumOne[i]+digitSumOne[j];
    }
    
}