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

算法记录

LeetCode 题目：

  地上有一个m行n列的方格，从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动，它每次可以向左、右、上、下移动一格（不能移动到方格外），也不能进入行坐标和列坐标的数位之和大于k的格子。

---

---

说明

一、题目

  请问该机器人能够到达多少个格子？

二、分析

• 简单的迷宫求解问题, 需要注意的是边界的判定,.
• 一个是下标之间的关系, 需要有一个函数进行计算, 数据量小也可以用一个表来存储对应关系.
• 而是需要输出的是整个算法中可到达的点的格数, 也就是整个队列的输出长度.

class Solution {
    int[][] direction = {{0,1},{0,-1},{1,0},{-1,0}};
    public int movingCount(int m, int n, int k) {
        if(k == 0) return 1;
        int ans = 0;
        Queue<Node> queue = new ArrayDeque();
        int[][]  visit = new int[m][n];
        queue.add(new Node(0,0));
        visit[0][0] = 2;
        while(!queue.isEmpty()) {
            Node node = queue.poll();
            ans++;
            for(int i = 0; i < 4; i++) {
                int x = node.x + direction[i][0];
                int y = node.y + direction[i][1];
                if(x < 0 || y < 0 || x >= m || y >= n) continue;
                if(visit[x][y] == 2) continue;
                if(sumIndex(x, y) > k) continue;
                visit[x][y] = 2;
                queue.add(new Node(x, y));
            }
        }
        return ans;
    }

    private int sumIndex(int x, int y) {
        int ret = 0;
        while(x > 0) {
            ret += x % 10;
            x = x / 10;
        }
        while(y > 0) {
            ret += y % 10;
            y = y / 10;
        }
        return ret;
    }

    private class Node{
        int x;
        int y;
        public Node(int x, int y) {
            this.x = x;
            this.y = y;
        }
    }
}

---

总结

熟悉广度优先搜索。