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

问题

题目:[剑指offer-机器人的运动范围]

思路

dfs+剪枝的常见思路。
注意,这个题是常规题目,上一道题目不太常规是因为它回溯之后,要恢复visited的未被访问的状态!!!
其实,正常的题目是不需要的。

忘了说的是,刚上来其实我是想直接数格子。但是数格子的问题是,有些格子合法但是你到不了!
比如,threshold = 9. (18, 0)合法,但是(19,0)不合法,到不了。但是(20,0)合法,此时也到不了。

代码

class Solution {
public:
    int movingCount(int threshold, int rows, int cols)
    {
        int cnt = 0;
        vector<vector<int>> visited( rows, vector<int>(cols, 0) );
        dfs( rows, cols, 0, 0, visited, threshold, cnt );
        return cnt;
    }
private:
    void dfs(int rows, int cols,
             int i, int j, vector<vector<int>>& visited,
             int threshold, int& cnt){
        if( i < 0 || i == rows || j < 0 || j == cols || visited[i][j] ) return;

        visited[i][j]=1;
        int sum = sum_digits(i) + sum_digits(j);
        if( sum <= threshold ){
            cnt++;
            dfs( rows, cols, i-1, j, visited, threshold, cnt );
            dfs( rows, cols, i+1, j, visited, threshold, cnt );
            dfs( rows, cols, i, j-1, visited, threshold, cnt );
            dfs( rows, cols, i, j+1, visited, threshold, cnt );
        }
    }
    int sum_digits(int val){
        stringstream ss;
        ss << val;
        string tmp;
        ss >> tmp;

        int sz = tmp.size();
        int ans = 0;
        for(int i = 0; i < sz; ++i){
            ans += (tmp[i] - '0');
        }
        return ans;
    }
};

你可能感兴趣的:(剑指offer)