2022-6-26 剑指offer-BFS

剑指 Offer 13. 机器人的运动范围

难度中等522收藏分享切换为英文接收动态反馈

地上有一个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。请问该机器人能够到达多少个格子？

class Solution {
    public int movingCount(int m, int n, int k) {
        int[][] dir={{-1,0},{1,0},{0,1},{0,-1}};
        Queue<int[]> queue=new  LinkedList<>();
        boolean[][] visited=new boolean[m][n];
        queue.offer(new int[]{0,0});
        visited[0][0]=true;
        int ans=0;
        while (!queue.isEmpty()){
            int[] temp=queue.poll();
            ans++;
            int x=temp[0];
            int y=temp[1];
            for  (int i=0;i<4;i++){
                int dx=x+dir[i][0];
                int dy=y+dir[i][1];
                if (dx>=0&&dx<m&&dy>=0&&dy<n&&!visited[dx][dy]&&(count(dx)+count(dy)<=k)){
                    queue.offer(new int[]{dx,dy});
                    visited[dx][dy]=true;
                }
            }
        }
        return ans;
    }


    public int count(int x){
        int sum=0;
        while (x>0){
            sum+=x%10;
            x/=10;
        }
        return sum;
    }
}

思路：BFS搜索。