剑指 Offer 13. 机器人的运动范围

题目：

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

示例 1：

输入：m = 2, n = 3, k = 1
输出：3

示例 2：

输入：m = 3, n = 1, k = 0
输出：1

提示：

1 <= n,m <= 100
0 <= k <= 20

题解：

我们将行坐标和列坐标数位之和大于 k 的格子看作障碍物，那么这道题就是一道很传统的搜索题目，我们可以使用广度优先搜索或者深度优先搜索来解决它。

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
    bool OK(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;
        return false;
    }
    int bfs(int m,int n,int k){
        int vis[m][n];
        memset(vis,0,sizeof(vis));
        queue<node>queue;
        queue.push(node(0,0));
        vis[0][0]=1;
        int ans=0;
        int dir[4][2]={0,1,1,0,0,-1,-1,0};
        while(!queue.empty()){
            node now = queue.front();
            queue.pop();
            ans++;
            for(int i=0;i<4;i++){
                int x,y;
                x= now.i+dir[i][0];
                y = now.j+dir[i][1];
                if(x>=0&&y>=0&&x<m&&y<n&&vis[x][y]==0&&OK(x,y,k)){
                    vis[x][y]=1;
                    queue.push(node(x,y));
                }
            }

        }
        return ans;
    }
    int movingCount(int m, int n, int k) {
        return bfs(m,n,k);
    }

private:
    struct node{
        int i,j;
        node(int i,int j){
            this->i = i;
            this->j = j;
        }
    };
};
int main() {
    int n,m,k;
    cin>>n>>m>>k;
    Solution solution;
    cout<<solution.movingCount(m,n,k);
}