剑指Offer-机器人的运动范围

傷城~ 2022-05-22 03:10 216阅读 0赞

## 题目描述 ##

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

**算法思路**

定义和方格大小一样的布尔值矩阵visited，用来标识路径是否已经到达过每个格子。

当首次到达某个格子（i,j）的时候，即visited\[i\*cols+j\]==false，如果满足约束条件，计数加1并从该格出发，分别从四个方向进行移动；若该格已经被访问过或者不满足约束条件，返回0。

**代码实现**

class Solution {
    private:
        int nsteps;
    public:
        Solution(){
            nsteps = 0;
        }
        int sum(int i, int j){
            int s = 0;
            while(i){
            	s += i%10;
            	i = i /10;
    		}
    		while(j){
    			s += j%10;
    			j = j /10;
    		}
    		return s;
        }
        int nCount(int i, int j, int threshold, int rows, int cols, bool* visited){
            int index = i*cols+j;
            if(i<0 || i>=rows || j<0 || j>=cols || sum(i,j)>threshold || visited[index])
                return 0;
            visited[index] = true;
            return  1+nCount(i+1,j,threshold,rows,cols,visited)
                    +nCount(i-1,j,threshold,rows,cols,visited)
                    +nCount(i,j+1,threshold,rows,cols,visited)
                    +nCount(i,j-1,threshold,rows,cols,visited);
        }
        int movingCount(int threshold, int rows, int cols)
        {
            bool *visited = (bool*)malloc((rows*cols)*sizeof(bool));
            for(int i = 0; i <rows*cols;++i)
                visited[i] = false;
            nsteps = nCount(0, 0, threshold, rows, cols, visited);
            return nsteps;
        }
    };