【每日打卡】Day12：机器人的运动范围 C++实现

地上有一个 mm 行和 nn 列的方格，横纵坐标范围分别是 0∼m−10∼m−1 和 0∼n−10∼n−1。

一个机器人从坐标0,0的格子开始移动，每一次只能向左，右，上，下四个方向移动一格。

但是不能进入行坐标和列坐标的数位之和大于 kk 的格子。

请问该机器人能够达到多少个格子？

样例1

输入：k=7, m=4, n=5

输出：20

样例2

输入：k=18, m=40, n=40

输出：1484

解释：当k为18时，机器人能够进入方格（35,37），因为3+5+3+7 = 18。
      但是，它不能进入方格（35,38），因为3+5+3+8 = 19。

注意:

1. 0<=m<=50
2. 0<=n<=50
3. 0<=k<=100

//
// Created by Dr.W on 2019/4/23.
//

#include <bits/stdc++.h>

using namespace std;
//创建方向数组，方便枚举四个方向
int dx[4] = {-1,0,1,0}, dy[4] = {0,1,0,-1}; 
//vis数组，标记数组元素是否访问过   
int vis[55][55];                               
int k,m,n,ans = 0;
//取数位函数
int getsum(int x){
    int sum = 0;
    while(x > 0){
        sum += x % 10;
        x /= 10;
    }
    return sum;
}
//深搜
void dfs(int x ,int y){
    //能搜到说明可被访问 vis置1
    vis[x][y] = 1;
    //可访问坐标数加一，也可以在最后循环统计vis数组中元素值为1的数量
    ans ++;
    //DFS出口
    if(x >= m || y >= n)
        return;
    for(int i = 0; i < 4; i ++){
        int nx = x + dx[i], ny = y + dy[i];
        if(nx >= 0 && ny >= 0 && nx < m && ny < n && getsum(nx) + getsum(ny) <= k && vis[nx][ny] != 1){
            dfs(nx,ny);
        }
    }
}

int solve(int k,int m,int n){
    dfs(0,0);
    //第一次提交忽略了数组为空的情况 应当考虑在内!
    if(m == 0 && n == 0)
        return 0;
    return ans;
}

int main(){
    cin >> k >> m >> n;
    memset(vis,0, sizeof(vis));
    cout << solve(k,m,n);
    return 0;
}