剑指 Offer 13. 机器人的运动范围
思路
参考 剑指 Offer 13. 机器人的运动范围( 回溯算法,DFS / BFS ,清晰图解)
1.数位和的增量公式
设 x 的数位和为 S(x) ,x+1 的数位和为 S(x+1)
若x%10==0,S(x+1)=S(x)-8,如x=19,S(x)=10,x+1=20,S(x+1)=2=S(x)-8;
若x%10!=0,S(x+1)=S(x)+1,如x=18,S(x)=9,x+1=19,S(x+1)=10=S(x)+1
2.机器人只需要选择向下或者向右走
因此最后的结果为
1 + dfs(i + 1, j, 数位和) + dfs(i, j + 1, 数位和);
代码
class Solution {
int m,n,k;
boolean [][]v;
public int movingCount(int m, int n, int k) {
this.m=m;
this.n=n;
this.k=k;
v=new boolean[m][n];
return dfs(0,0,0,0);
}
public int dfs(int i,int j,int si,int sj){
if(i>=m||j>=n||si+sj>k||v[i][j])return 0;
v[i][j]=true;
return 1+dfs(i+1,j,(i+1)%10==0?si-8:si+1,sj)+dfs(i,j+1,si,(j+1)%10==0?sj-8:sj+1);
}
}