跳至主要內容

13-机器人的运动范围

daipeng大约 2 分钟

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

题目中的可到达的格子是可以移动到的格子,假如k=1,那么(10,0)虽然满足不大于k,但是由于(9,0),(11,0),(10,1)都无法到达,所以(10,0)也是无法到达的。

那么我们参照12-矩阵中的路径的解法,依然是走迷宫的思路,我们找一个点,开始走,如果下一个格子可以通过,那么继续找这个格子的四周的格子,然后循此往复,直到再也没有可以走的格子。这实际就是深度优先搜索的方法。

以下面的代码为例

  public int movingCount(int k, int rows, int cols)
    {
        if(k == 0){
            return 1;
        }
        boolean[] visited = new boolean[rows*cols];
        int count = movingCount(rows,cols,k,0,0,visited);
        return count;

    }
    public int movingCount(int rows,int cols,int k,int row, int col,boolean[] visited){
        int count = 0;
        int index = row*cols+col;
        // 如果格子在矩阵范围内 同时没有访问过 同时符合要求
        if(row >= 0 && col >= 0 && row < rows && col < cols && !visited[index] && (getDigitSum(row)+getDigitSum(col)) <= k){
            //进行选择 标记已经访问过了
            visited[index] = true;
            //继续访问结点周围的结点
            count = 1 + movingCount(rows,cols,k,row-1,col,visited)
                             +movingCount(rows,cols,k,row+1,col,visited)
                             +movingCount(rows,cols,k,row,col-1,visited)
                             +movingCount(rows,cols,k,row,col+1,visited);
                    
                              
        }
        return count;
    }
    public int getDigitSum(int num){
        int sum = 0;
        int rem = 0;
        while(num != 0 ){
            rem = num % 10;
            sum += rem;
            num /= 10;
        }
        return  sum;
    }