机器人的运动范围

题目描述

地上有一个m行和n列的方格。一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格，但是不能进入行坐标和列坐标的数位之和大于k的格子。 例如，当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。但是，它不能进入方格（35,38），因为3+5+3+8 = 19。请问该机器人能够达到多少个格子？

思路

这是一个回溯问题，从(0，0)开始判断当前格子行坐标和列坐标之和是否大于K，如果大于则将当前的count返回，若不大于则将flag矩阵中的对应位置的标志更改，表示走过了相应的格子，并且将count + 1。再依次向这个格子的上下左右四个方向去找，并将返回的结果给count。

代码

public int movingCount(int threshold, int rows, int cols)
    {
        
        if(threshold <= 0){
        	return 0;
        }
        int count = 0;
        boolean[][] flagMatrix = new boolean[rows][cols];
        count = judge(threshold, rows, cols, flagMatrix, 0, 0);
        
        return count;
    }
	
	public int judge(int k, int rows, int cols, boolean[][] flagMatrix, int i, int j) {
		int count = 0;
		
		if(i<0 || j<0 || i>= rows || j>= cols){
			return count;
		}
		
		if(!flagMatrix[i][j] && check(i, j) <= k){
			count = count + 1;
			flagMatrix[i][j] = true;
			count = count
					+ judge(k, rows, cols, flagMatrix, i, j-1)
					+ judge(k, rows, cols, flagMatrix, i, j+1)
					+ judge(k, rows, cols, flagMatrix, i-1, j)
					+ judge(k, rows, cols, flagMatrix, i+1, j);
		}
		
		return count;
	}
	
	public int check(int i, int j) {
		
		int res = 0;
		while(i!=0){
			res += i % 10;
			i = i / 10;
		}
		
		while(j!=0){
			res += j % 10;
			j = j / 10;
		}
		
		return res;
	}
	
	@Test
	public void test() {
		System.out.println(movingCount(5, 10, 10));
	}