剑指Offer - 面试题13：机器人的运动范围

题目

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

分析

dfs + 回溯
我们可以设置一个二维数组用来区别该位置是否计算过。初始化成0，如果计算过就设置为-1。然后从(0,0)开始，如果(0,0)都不符合那就所有位置都不符合了。然后与依次向上、下、左、右遍历。如下k=5时

绿色表示机器人可以到达的位置，
蓝色表示机器人不可以到达的位置，但是坐标符合。
空白就是坐标不符合

分析上图，我们发现9到10，19到20是零界点。

再测试k=11

总结上图。我们可以发现从(0,0)开始向下和向右移动就可以到达所有绿色区域，

C

#include<stdio.h>
#include<stdlib.h>

int getDigitSum(int num);
int movingCount(int rows, int cols, int k);
int movingCountCore(int** test, int rows, int cols, int i, int j, int k);

int main()
{
	int k = 5;

	int ret = movingCount(80, 100, k);

	if (-1 == ret)
	{
		printf("行或列小于零\n");
	}
	else
	{
		printf("机器人能够到达%d个格子\n", ret);
	}

	return 0;
}

int movingCount(int rows, int cols, int k)
{
	if (rows <= 0 || cols <= 0)
	{
		return -1;
	}

	int** test = (int**)calloc(rows * cols, sizeof(int));//申请一个标志数组

	if (NULL == test)
	{
		perror("申请空间失败");
	}

	int count = movingCountCore(test, rows, cols, 0, 0, k);

	free(test);//释放test数组

	return count;
}


int movingCountCore(int** test, int rows, int cols, int i, int j, int k)
{
	//行列超出边界，或者test[i][j]的值已经被修改，或者行坐标和列坐标的数位之和大于k的格子
	if (i >= rows || j >= cols || *((int*)test + i * cols + j) != 0 || (getDigitSum(i) + getDigitSum(j)) > k)
	{
		return 0;
	}

	*((int*)test + i * cols + j) = -1;

	//依次向下、右遍历
	return 1+ movingCountCore(test, rows, cols, i + 1, j, k)
		+ movingCountCore(test, rows, cols, i, j + 1, k);
}

int getDigitSum(int num)//坐标的位数之和
{
	int sum = 0;

	while (num != 0)
	{
		sum += num % 10;
		num /= 10;
	}
	return sum;
}

分析一下坐标的位数和：

当 0 <= i <= 9时，位数和就是本身
当 10 <= i <=19时，位数和是1+个位数
当 11 <= i <= 29时，位数和是2+个位数
…
位数和 = i %10 + 个位数
那么我们就可以省去getDigitSum()函数了。

优化后：

C

#include<stdio.h>
#include<stdlib.h>

int movingCount(int rows, int cols, int k);
int movingCountCore(int** test, int rows, int cols, int i, int j, int k);

int main()
{
	int k = 5;

	int ret = movingCount(80, 100, k);

	if (-1 == ret)
	{
		printf("行或列小于零\n");
	}
	else
	{
		printf("机器人能够到达%d个格子\n", ret);
	}

	return 0;
}

int movingCount(int rows, int cols, int k)
{
	if (rows <= 0 || cols <= 0)
	{
		return -1;
	}

	int** test = (int**)calloc(rows * cols, sizeof(int));//申请一个标志数组

	if (NULL == test)
	{
		perror("申请空间失败");
	}

	int count = movingCountCore(test, rows, cols, 0, 0, k);

	free(test);//释放test数组

	return count;
}


int movingCountCore(int** test, int rows, int cols, int i, int j, int k)
{
	//行列超出边界，或者test[i][j]的值已经被修改，或者行坐标和列坐标的数位之和大于k的格子
	if (i >= rows || j >= cols || *((int*)test + i * cols + j) != 0 || (i % 10 + i / 10 + j % 10 + j / 10) > k)
	{
		return 0;
	}

	*((int*)test + i * cols + j) = -1;

	//依次向下、右遍历
	return 1+ movingCountCore(test, rows, cols, i + 1, j, k)
		+ movingCountCore(test, rows, cols, i, j + 1, k);
}

本章完！