题目链接
题目描述
地上有一个m行n列的方格,从坐标 [0,0] 到坐标 [m-1,n-1] 。一个机器人从坐标 [0, 0] 的格子开始移动,它每次可以向左、右、上、下移动一格(不能移动到方格外),也不能进入行坐标和列坐标的数位之和大于k的格子。例如,当k为18时,机器人能够进入方格 [35, 37] ,因为3+5+3+7=18。但它不能进入方格 [35, 38],因为3+5+3+8=19。请问该机器人能够到达多少个格子?
示例 1:
输入:m = 2, n = 3, k = 1
输出:3
示例 2:
输入:m = 3, n = 1, k = 0
输出:1
提示:
1 <= n,m <= 100
0 <= k <= 20
解题思路
一开始想着用暴力,双重for循环判断哪些格子符合要求,但是这么判断的话,明显有一些格子是不能达到的,比如m=16 n=8,k=4,不光有前几个元素(10,0)这个格子也满足题意,但是与机器人的起点并不连通
然后就考虑染色法,先判断哪些格子可以满足,然后从起点(0,0)进入,看看能染到那个格子就说明最多能到达哪个位置
染色法
但是这里有一点要注意,当染色时如果这个位置的值已经取到的话就把这个位置置零,否则会重复取值,具体原因可以看一下下面的图例
0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | |
0 | 1 | 1 | 1 | 1 | 1 | 0 | 0 | 0 |
1 | 1 | 1 | 1 | 1 | 0 | 0 | 0 | 0 |
2 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
3 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
4 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
5 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
6 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
7 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
8 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
9 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
10 | 1 | 1 | 1 | 1 | 0 | 0 | 0 | 0 |
11 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
12 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
13 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
14 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
15 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
这是一开始的初始情况,如果不将选过的元素置零的话,遍历顺序为
0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | |
0 | 1 | 2 | 3 | 4 | 5 | 0 | 0 | 0 |
1 | 12 | 9/13 | 7/10/14 | 6/8/11/15 | 0 | 0 | 0 | 0 |
2 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
3 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
4 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
5 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
6 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
7 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
8 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
9 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
10 | 1 | 1 | 1 | 1 | 0 | 0 | 0 | 0 |
11 | 1 | 1 | 1 | 0 | 0 | 0 | 0 | 0 |
12 | 1 | 1 | 0 | 0 | 0 | 0 | 0 | 0 |
13 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
14 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
15 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
如图,当遍历到第二行的时候就会有元素重复被遍历,所以必须将已经算完的元素置零,置零以后不用怕会有元素漏下,因为肯定不会有元素漏下。
题解
染色法
class Solution {
public:
int g[105][105];
int cal(int x){
int sum=0;
while(x>0){
sum+=x%10;
x/=10;
}
return sum;
}
int ans=0;
void dfs(int x,int y,int m,int n){
if(g[x][y]==0||x>m||y>n) return ;
else{
g[x][y]=0;
ans++;
}
dfs(x,y+1,m,n);
dfs(x+1,y,m,n);
}
int movingCount(int m, int n, int k) {
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(cal(i)+cal(j)<=k){
cout<<i<<" "<<cal(i)<<" "<<j<<" "<<cal(j)<<endl;
g[i][j]=1;
}
}
}
dfs(0,0,m,n);
return ans;
}
};