机器人的运动范围

题目:

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

 

示例 1:

输入:m = 2, n = 3, k = 1
输出:3

示例 2:

输入:m = 3, n = 1, k = 0
输出:1

思路:

运用广度有优先搜索。设定队列保存待处理的数。设定一个数组保存已经处理过的数,达到除重的目的。

代码:

import java.util.LinkedList;
import java.util.Queue;

class Solution {
    public int movingCount(int m, int n, int k) {
        if(k==0){
            return 1;
        }
        int res=1;
        Queue<int[]> queue=new LinkedList<>();
        queue.offer(new int[]{0,0});
        int [][] nums=new int[m][n];

        while(!queue.isEmpty()){
            int[] cell=queue.poll();
            int y=cell[0],x=cell[1];

            if(y+1<m){
                if(getNum(y+1)+getNum(x)<=k&&nums[y+1][x]!=1){
                    queue.offer(new int[]{y+1,x});
                    nums[y+1][x]=1;
                    res++;
                }
            }
            if(x+1<n){
                if(getNum(x+1)+getNum(y)<=k&&nums[y][x+1]!=1){
                    queue.offer(new  int[]{y,x+1});
                    nums[y][x+1]=1;
                    res++;

                }
            }
        }
        return res;
    }
    public int getNum(int num) {
        int res=0;
        while (num!=0){
            res+=num%10;
            num=num/10;
        }
        return res;
    }
}

  

posted @ 2021-03-08 13:47  silence_bug  阅读(69)  评论(0)    收藏  举报