题目链接----------------------->面试题13. 机器人的运动范围
很简单的一道搜索题,之前都是用递归做的,这次用栈来实现dfs,其实本质是一样的,帖下代码:
int movingCount(int m, int n, int k) {
int cnt = 1,sum;
int fxy[4][2]={{1,0},{-1,0},{0,1},{0,-1}};
int vis[m][n];
memset(vis,0,sizeof(vis));
stack<pair<int,int>>s;
pair<int,int>p(0,0);
s.push(p);
vis[0][0] = 1;
while(!s.empty())
{
p = s.top();
s.pop();
for(int i = 0;i < 4;i++)
{
sum = 0;
int x = p.first + fxy[i][0];
int y = p.second + fxy[i][1];
int num_1 = x,num_2 = y;
while(num_1 || num_2)
{
if(num_1)
{
sum += num_1 % 10;
num_1 /= 10;
}
if(num_2)
{
sum += num_2 % 10;
num_2 /= 10;
}
}
if(x >= 0 && x < m && y >= 0 && y < n && !vis[x][y] && sum <= k)
{
cnt++;
pair<int,int>node(x,y);
s.push(node);
vis[x][y] = 1;
}
}
}
return cnt;
}