【剑指offer】机器人的运动范围

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

思路：直接利用dfs来做，然后加上里面的条件判断就行了，其实还蛮简单的。一下是我AC的代码

#include <iostream>
#include <string.h>
#include <vector>
#include <sstream>
#include <algorithm>
#include <stack>
using namespace std;
char field[500][500] = {};
//四个移动方向
int dx1[4] = { 1, 0, -1, 0 };
int dy1[4] = { 0, 1, 0, -1 };
int ans = 0;

void dfs(int x, int y,int rows,int cols, int k)
{
	//将当前的方格替换为.，表示已经访问过了
	field[x][y] = '.';
	for (int i = 0; i < 4; i++)
	{
		int newx = x + dx1[i];
		int newy = y + dy1[i];
		int tx = newx, ty = newy;
		int tsum = 0;
		while (tx > 0)
		{
			tsum += tx % 10;
			tx /= 10;
		}
		while (ty > 0)
		{
			tsum += ty % 10;
			ty /= 10;
		}
		if (0 <= newx && newx < rows && 0 <= newy && newy < cols && field[newx][newy] == '*' && (tsum <= k))
		{
			ans++;
			dfs(newx, newy, rows, cols, k);
		}
	}
}

int movingCount(int k, int rows, int cols)
{
	memset(field, '*', sizeof(field));
	ans = 1;
	dfs(0, 0, rows, cols, k);
	return ans;
}

int main()
{
	int c = movingCount(5, 10, 10);
	cout << c << endl;
}