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

剑指 Offer 13. 机器人的运动范围
地上有一个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

我觉得这道题题目表达的不清楚,应该补充:能省则省,这样就没必要走回头路,也就是没必要 去访问已经被访问过的格子(如果允许重复访问的话,结果应该是正无穷吧……);还应该补充,路径要连贯。

思路1:广度优先遍历

看下面的代码结构很像二叉树的按层遍历(只是这里“节点”被放进queue的判断条件不再是“左右孩子不为空”,而是 满足此处题意的条件——不超出整个面板的范围 && 横纵坐标数字和<=k && 还没有进过queue, 这个判断条件在下面的代码中被反着表述了出来)。

class Solution {
    public int movingCount(int m, int n, int k) {
        int[] dx = {0,1};
        int[] dy = {1,0};
        boolean[][] visited = new boolean[m][n];
        Queue<int[]> queue = new LinkedList<>();
        int[] cur = new int[]{0,0};
        queue.offer(cur);
        visited[0][0]=true;
        int ans=1;
        while(!queue.isEmpty()){
            cur = queue.poll();
            for(int i=0;i<2;i++){
                int x = cur[0]+dx[i];
                int y = cur[1]+dy[i];
                if(x<0||x>=m||y<0||y>=n||(get(x)+get(y))>k||visited[x][y]){
                    continue;
                }

                queue.offer(new int[]{x,y});
                visited[x][y]=true;
                ans++;
            }
        }
        return ans;
        
    }
    public int get(int x){
        int res=0;
        while(x!=0){
            res+=x%10;
            x/=10;
        }
        return res;
    }
}

思路2:深度优先遍历

根据上面的广度优先搜索,可以写出深度优先搜索的代码如下。
注意:因为事先想好了,每次决策都是“向下”或“向右”,所以dsf递归函数的base case 只需照顾x,y的上限即可,而不用操心下限的情况。
有意思的是,这里的深度优先搜索并不用恢复现场,因为本题希望求有多少格子是机器人能走的

  int count=0;
    public int movingCount(int m, int n, int k) {
        count++;
        boolean[][] visited = new boolean[m][n];
        visited[0][0]=true;
        dsf(m,n,0,0,k,visited);
        return count;
    }
    public void dsf(int m, int n, int x, int y, int k,boolean[][] visited){
        if(x==m&&y==n){
            return;
        }
     if(x+1<m && (get(x+1)+get(y))<=k && !visited[x+1][y]){
            count++;
            visited[x+1][y]=true;
            dsf(m,n,x+1,y,k,visited);

        }
        if(y+1<n && (get(x)+get(y+1))<=k  && !visited[x][y+1]){
            count++;
            visited[x][y+1]=true;
            dsf(m,n,x,y+1,k,visited);
        }
    }
 public int get(int x){
        int res=0;
        while(x!=0){
            res+=x%10;
            x/=10;
        }
        return res;
    }

你可能感兴趣的:(数据结构与算法,机器人,算法)