LeetCode 面试题13. 机器人的运动范围（4.8打卡）

题目

面试题13. 机器人的运动范围
地上有一个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
示例 1：

输入：m = 3, n = 1, k = 0
输出：1

小记

第一次方法遇到了问题，一直可以在eclipse和LeetCode测试界面正确运行，但是提交一直出来错误答案，找不到原因，最后放在了讨论区，LeetCode大神救了我，感谢，大写的感谢！

代码

//第一次方法，提交一直过不了，在LeetCode讨论区遇到了大神，指出了问题，很高兴遇到LeetCode的大神，哈哈
class Solution {

	static int ans=0;
	static int[] sM;
	static int[] sN;
	static int[][] flag;
	static int Mm,Nn,Kk;
	static int[] dx= {0,0,1,-1};
	static int[] dy= {-1,1,0,0};
	
	public static int movingCount(int m, int n, int k) {
        ans=0;//此处每次运行需要将ans清零，否则ans会和之前的运行有关，结果会出错
		if(m==0||n==0)
			return 0;
        if(m+n<k)
			return n*m;
		//辅助数组，保存每个i,j对应的各位数字之和
		//广度优先搜索
		//BFS
		Mm=m;
		Nn=n;
		Kk=k;
		sM=new int[m];
		sN=new int[n];
		flag=new int[m][n];
		for(int i=0;i<m;i++) {//初始化sM
			int t=i;int y=0;int sum=0;
			while(t>0) {
				y=t%10;
				sum+=y;
				t=t/10;
			}
			sM[i]=sum;
		}
		for(int i=0;i<n;i++) {//初始化sN
			int t=i;int y=0;int sum=0;
			while(t>0) {
				y=t%10;
				sum+=y;
				t=t/10;
			}
			sN[i]=sum;
		}
		ans++;
		flag[0][0]=1;
		BFS(0,0);
		return ans;
    }
	
	private static void BFS(int mi, int ni) {
		if(mi>Mm||ni>Nn)
			return ;
		for(int i=0;i<4;i++) {
			int x=dx[i]+mi;int y=dy[i]+ni;
			if(x>=0&&y>=0&&x<Mm&&y<Nn&&flag[x][y]==0) {
				if(sM[x]+sN[y]<=Kk) {
					flag[x][y]=1;ans++;
					BFS(x,y);
					}
			}
		}
		return ;
	}
}

/*作者：宇zzZ
链接：https://leetcode-cn.com/circle/discuss/HKD0df/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/

//第二次写的代码，运行比较顺利
class Solution {
    public static int movingCount(int m, int n, int k) {
		int ans=0;
		int[] hm=new int[m];
		int[] hn=new int[n];
		//初始化
		inite(hm,m);
		inite(hn,n);
		int[][] flag=new int[m][n];
		ans=BFS(0,0,m,n,k,hm,hn,flag);
		return ans;
    }

	private static void inite(int[] a, int m) {
		for(int i=0;i<m;i++) {
			int t=i;
			int y=0;int sum=0;
			while(t>0) {
				y=t%10;
				sum+=y;
				t=t/10;
			}
			a[i]=sum;
		}
	}

	private static int BFS(int i, int j, int m, int n, int k, int[] hm, int[] hn, int[][] flag) {
		if(i<0||j<0||i>=m||j>=n||flag[i][j]==1)
			return 0;
		else{
			if(hm[i]+hn[j]<=k) {
				//System.out.println(i+"_"+j);
				flag[i][j]=1;
				return 1+BFS(i+1,j,m,n,k,hm,hn,flag)+BFS(i,j+1,m,n,k,hm,hn,flag)+BFS(i-1,j,m,n,k,hm,hn,flag)+BFS(i,j-1,m,n,k,hm,hn,flag);
			}
			return 0;
		}
	}
}

//大神的0ms代码（搬运工的日常）
class Solution {
    int m, n, k;
    boolean[][] visited;
    
    public int movingCount(int m, int n, int k) {
        this.m = m; this.n = n; this.k = k;
        this.visited = new boolean[m][n];
        return dfs(0, 0, 0, 0);
    }

    public int dfs(int i, int j, int si, int sj) {
        if(i < 0 || i >= m || j < 0 || j >= n || k < si + sj || visited[i][j]) return 0;
        visited[i][j] = true;
        return 1 + dfs(i + 1, j, (i + 1) % 10 != 0 ? si + 1 : si - 8, sj) + dfs(i, j + 1, si, (j + 1) % 10 != 0 ? sj + 1 : sj - 8);
    }
}