剑指offer-66.机器人的运动范围

题目描述

地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

解题思路

思路：回溯法；

用一个函数来计算行坐标和列坐标的数位之和；

用一个函数判断终止条件：包括三种情况：越界、重复、行坐标和列坐标的数位之和超过k

代码实现：

public class Solution {
    public int movingCount(int threshold, int rows, int cols)
    {
        if(threshold < 0 || rows <= 0 && cols <= 0){
            return 0;
        }
        int[] visited = new int[rows*cols];//表示是否访问
       int ans = dfs(threshold,rows,cols,0,0,visited);
        return ans;
    }
    //回溯法
    int dfs(int threshold,int rows,int cols,int i,int j,int[] visited){
        int count = 0;
        int index = i*cols + j;
       if(check(threshold,rows,cols,i,j,visited)){
           visited[index] = 1;
           // 向上下左右寻找
           count = 1+dfs(threshold,rows,cols,i-1,j,visited)
               +dfs(threshold,rows,cols,i+1,j,visited)
               +dfs(threshold,rows,cols,i,j-1,visited)
               +dfs(threshold,rows,cols,i,j+1,visited);
       }
        return count;
    }
    //判断是否符合要求
    boolean check(int threshold,int rows,int cols,int i,int j,int[] visited){
        if(i >= 0 && i<rows && j >=0 && j<cols && getSum(i)+getSum(j)<=threshold && visited[i*cols+j]==0)
            return true;
        return false;
    }
    //计算坐标的数位之和
    int getSum(int num){
        int s = 0;
        while(num != 0){
            s+= num%10;
            num/=10;
        }
        return s;
    }
}

 效率：