leetcode之机器人运动范围

地上有一个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。请问该机器人能够到达多少个格子?
输入:m = 2, n = 3, k = 1
输出:3
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ji-qi-ren-de-yun-dong-fan-wei-lcof
  • 一般遇到需要遍历整个矩阵找到答案的时候,可以考虑使用深度优先搜索或者广度优先搜索
  • 首先我们使用深度优先搜索,这里需要注意的是,我们既然要统计可以到达的格子数,那么一定要对已经遍历过的格子做个标记,不然后面再来搜索的时候会重复计算;这里有个重要的地方就是我们一开始就设置为从(0,0)开始走,所以我们只需要计算向右和向下的部分,不需要i-1和j-1;
class Solution {
    int m;
    int n;
    int k;
    int sum;
    //这里使用set集合也是可以的,但是时间消耗比较多,相对来说还是操作数组比较方便
    // Set<String> set = new HashSet<>();
    //创建一个m*n大小的数组,用来标记已经访问过的路径;
    boolean[][] visited;
    public int movingCount(int m, int n, int k) {
    //使用这种全局变量的方式就不用带着参数到处跑了
        this.m = m;
        this.n = n;
        this.k = k;
        visited = new boolean[m][n];
        //利用深度优先搜索
        dfs(0,0);
        return sum;
    }
    public void dfs(int i,int j){
    //防止数组下标越界
        if(i < 0 || j < 0 || i >= m || j >= n || visited[i][j] == true){
            return;
        }
        int count = 0;
     	//以下是计算数位和
        int im = i;
        int jn = j;
        while(im > 0){
            count += (im % 10);
            im /= 10;
        }
        while(jn > 0){
            count += (jn % 10);
            jn /= 10;
        }
        if(count > k){
            return;
        }
        // String key = i+" "+j;
        // if(!set.add(key)){
        //     return;
        // }
        visited[i][j] = true;
        sum++;
        //向下
        dfs(i+1,j);
        // dfs(i-1,j);
        //向右
        dfs(i,j+1);
        // dfs(i,j-1);
    }
}
  • 以下是广度优先搜索,我们一般使用队列去实现广度优先搜索
class Solution {
    public int movingCount(int m, int n, int k) {
        Deque<int[]> queue = new LinkedList<>();
        boolean[][] visited = new boolean[m][n];
        // Set<String> set = new HashSet<>();
        queue.add(new int[]{0,0});
        int count = 0;
        while(queue.size() > 0){
            int sum = 0;
            int[] arr = queue.poll();
            int i = arr[0];
            int j = arr[1];
            int im = i;
            int jn = j;
            if(i < 0 || j < 0 || i >= m || j >= n || visited[i][j] == true){
                continue;
            }
            while(im > 0){
                sum += (im % 10);
                im /= 10;
            }
            while(jn > 0){
                sum += (jn % 10);
                jn /= 10;
            }
            if(sum > k){
                continue;
            }
            // String key = i + " " + j;
            // if(!set.add(key)){
            //     continue;
            // }
            visited[i][j] = true;
            count++;
            queue.add(new int[]{i+1,j});
            queue.add(new int[]{i,j+1});
        }                                                                            
        return count;
    }
}

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/harryshumxu/article/details/105396286