剑指刷题 机器人的运动

题目描述

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

 

解题思路：用回溯法，如果当前格子可以走的话，那么再看它上下左右的格子是否能走。

class Solution {
private:bool *visited;
public:
	int movingCount(int threshold, int rows, int cols)
	{
		visited = new bool[rows*cols];
		memset(visited, 0, rows*cols);
		int count = walk(0, 0, threshold, rows, cols);
		delete[] visited;
		return count;

	}
	int walk(int r, int c, int threshold, int rows, int cols)
	{
		int count = 0;
		if (r < 0 || r >= rows || c < 0 || c >= cols)
			return 0;
		if (can_go(r, c, threshold, rows, cols))
		{
			visited[r*cols + c] = true;
			count= 1 + walk(r + 1, c, threshold, rows, cols)
				+ walk(r - 1, c, threshold, rows, cols)
				+ walk(r , c+1, threshold, rows, cols)
				+ walk(r , c-1, threshold, rows, cols);

		}
		return count;
	}
	bool can_go(int r, int c, int threshold, int rows, int cols)
	{
		if (visited[r*cols + c])
			return false;
		int plus = 0;
		while (r != 0)
		{
			plus+=(r % 10);
			r /= 10;
		}
		while (c != 0)
		{
			plus +=  (c % 10);
			c /= 10;
		}
		if (plus <= threshold)
			return true;
		return false;
	}
};

还有一道类似的题   矩阵的路径

题目描述

请设计一个函数，用来判断在一个矩阵中是否存在一条包含某字符串所有字符的路径。路径可以从矩阵中的任意一个格子开始，每一步可以在矩阵中向左，向右，向上，向下移动一个格子。如果一条路径经过了矩阵中的某一个格子，则之后不能再次进入这个格子。 例如 a b c e s f c s a d e e 这样的3 X 4 矩阵中包含一条字符串"bcced"的路径，但是矩阵中不包含"abcb"路径，因为字符串的第一个字符b占据了矩阵中的第一行第二个格子之后，路径不能再次进入该格子。

class Solution {
private:
	bool *visited;
public:
	bool hasPath(char* matrix, int rows, int cols, char* str)
	{
		if (matrix == NULL || rows < 1 || cols < 1 || str == NULL)
			return false;
		visited = new bool[rows*cols];
		memset(visited, 0, rows*cols);
		bool flag = false;
		for (int r = 0; r < rows; ++r)
		{
			for (int c = 0; c < cols; ++c)
			{
				if (matrix[r*cols+c]==str[0])
				{
					memset(visited, 0, rows*cols);
					flag= walk(r, c, matrix, rows, cols, str,0);
					if (flag)
						return true;
				}
			}
		}

		delete[] visited;
		return false;

	}
	bool walk(int r, int c, char* matrix, int rows, int cols, char* str,int index)
	{
		int count = 0;
		if (r < 0 || r >= rows || c < 0 || c >= cols)
			return 0;
		if (can_go(r, c, matrix, rows, cols, str,index))
		{
			visited[r*cols + c] = true;
			if (index +1 == strlen(str))
				return true;
			return walk(r + 1, c, matrix, rows, cols, str, index+1)
				|| walk(r - 1, c, matrix, rows, cols, str, index + 1)
				|| walk(r, c + 1, matrix, rows, cols, str, index + 1)
				|| walk(r, c - 1, matrix, rows, cols, str, index + 1);

		}
		return false;
		return count;
	}
	bool can_go(int r, int c, char* matrix, int rows, int cols, char* str,int index)
	{
		if (visited[r*cols + c])
			return false;
		if (matrix[r*cols + c] == str[index])
			return true;
		return false;

	}

};