劍指Offer學習-面試題13:機器人的運動範圍

	/**
     * 機器人的運動範圍
     * 地上有一個m行n列的方格,一個機器人從座標(0,0)開始移動,它每次可以向左,右,上,下移動一格,
     * 但不能進入行座標和列座標的數位之和大於k的格子。
     * 例如 k=18 (35,37) 3+5+3+7=18 可以進入 (35,38) 3+5+3+8 = 19 不能進入
     * <p>
     *
     * @param m
     * @param n
     * @param k
     * @return
     */
    public int movingCount(int m, int n, int k) {
        boolean[][] visited = new boolean[m][n];
        int count = movingCountCore(m, n, k, 0, 0, visited);
        return count;
    }

    private int movingCountCore(int m, int n, int k, int x, int y, boolean[][] visited) {
        if (x < 0 || x >= m || y < 0 || y >= n || visited[x][y] || (getDigitSum(x) + getDigitSum(y)) > k) return 0;
        int count = 0;
        visited[x][y] = true;
        count++;
        count += movingCountCore(m, n, k, x + 1, y, visited);
        count += movingCountCore(m, n, k, x - 1, y, visited);
        count += movingCountCore(m, n, k, x, y + 1, visited);
        count += movingCountCore(m, n, k, x, y - 1, visited);
        return count;

    }

    private int getDigitSum(int num) {
        int sum = 0;
        while (num > 0) {
            sum += num % 10;
            num /= 10;
        }
        return sum;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章