问题
思路
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;
}
};