剑指offer——机器人的运动范围(13题)

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

此题与12题一样,同样也要运用回溯法,只不过每遍历一个格子时,还要进行一次计算判决。代码如下所示:

#include<vector>
#include<iostream>
#include<string>

using namespace std;

static int k;//指定的数

bool isOk(int x, int y, int k) {
	int sum = 0;
	while (x) {
		sum += x % 10;
		x /= 10;
	}
	while (y) {
		sum += y % 10;
		y /= 10;
	}
	if (sum <= k)
		return true;
	else
		return false;
}

int numberOfArea(int& rows, int& cols, int x, int y, vector<vector<bool>>& visited, int& cnt) {
	if (x < 0 || x >= rows || y<0 || y>cols || !visited[x][y] || !isOk(x, y, k))
		return cnt;

	//如果该结点没有访问且isOK
	visited[x][y] = true;
	cnt++;

	numberOfArea(rows, cols, x - 1, y, visited, cnt);//上
	numberOfArea(rows, cols, x + 1, y, visited, cnt);//下
	numberOfArea(rows, cols, x, y - 1, visited, cnt);//左
	numberOfArea(rows, cols, x, y + 1, visited, cnt);//右

	return cnt;
}

猜你喜欢

转载自blog.csdn.net/TT_love9527/article/details/82429577