回溯法,做的时候对 递归的结束条件 想不出,1、觉得一个节点的上下左右遍历一定要满足一定顺序,不然会丢项。2、终止条件为一个点不满足数位之和<=K,之后它的上下左右都不访问了,上右不看没毛病,那左下呢,怕后来给漏了,换个顺序测试了一下,都通过了测试,这个现象的本质还没弄懂……
#include <vector>
#include <iostream>
using namespace std;
//剑指 Offer 13. 机器人的运动范围
class Solution {
public:
int movingCount(int m, int n, int k) {
vector<vector<bool>> visited;
visited.resize(m);
for (int i = 0; i < visited.size(); i++) {
visited[i].resize(n);
}
for (int i = 0; i < visited.size(); i++) {
for (int j = 0; j < visited[0].size(); j++) {
visited[i][j] = false;
}
}
return Rec_Core(m - 1, n - 1, k, 0, 0, visited);//m和n的意义换了一下
}
//m和n的意义变换一下,题目中——从坐标 [0,0] 到坐标 [m-1,n-1](m行n列),变为坐标极限为(0,0)到(m,n)
int Rec_Core(int m, int n, int k, int row, int col, vector<vector<bool>>& visited) {//递归核心
if (BorJud(m, n, row, col) && DigitSum(row, col) <= k && visited[row][col] != true) {
visited[row][col] = 1;
//int res = 1 + Rec_Core(m, n, k, row - 1, col, visited) +
// Rec_Core(m, n, k, row, col - 1, visited) +
// Rec_Core(m, n, k, row + 1, col, visited) +
// Rec_Core(m, n, k, row, col + 1, visited);//这里产生一个疑问:现在的遍历顺序是下左上右,如果换一个顺序会不会有问题呢,试一下
int res = 1 + Rec_Core(m, n, k, row, col + 1, visited) + //结果是也通过了测试,现象值得深思
Rec_Core(m, n, k, row + 1, col, visited) +
Rec_Core(m, n, k, row, col - 1, visited) +
Rec_Core(m, n, k, row - 1, col, visited);
return res;
}
return 0;
}
bool BorJud(int m, int n, int row, int col) {
if (m < 0 || n < 0) {//输入错误
cout << "error input!";
return false;
}
if (row<0 || col<0 || row>m || col>n) {//坐标超出边界
return false;
}
return true;
}
int DigitSum(int row, int col) {
int res = row / 10 + row % 10 + col / 10 + col % 10;//题中说 1 <= n,m <= 100,上面做了m,n的意义变换,所以m,n属于[0,99]
return res;
}
};
int main(void) {
Solution solution;
int res = solution.movingCount(3,1,0);
cout << res;
return 1;
}