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

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

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

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

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

样例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

评测来自 ?   https://www.acwing.com/

bfs 做这道题,筛出不符合条件的点

class Solution {
public:
    typedef pair<int,int> PII;
    queue<PII> q;
    int vis[51][51];
    int count = 1;
    
    int fun(int x){
        int s = 0;
        while(x){
            s += x % 10;
            x /= 10;
        }
        return s;
    }
    
    void bfs(int target,int row,int col){
        int dx[] = {0,0,1,-1};
        int dy[] = {1,-1,0,0};
        while(q.size()){
            auto t = q.front();
            for (int i = 0; i < 4; ++i){
                int h = t.first + dy[i];
                int w = t.second + dx[i];
                if (vis[h][w]) continue;
                if(h >= 0 && h < row && w >= 0 && w < col && fun(h) + fun(w) <= target){
                    q.push({h,w});
                    count ++;
                    vis[h][w] = 1;
                }
            }
            q.pop();
        }
    }
    
    int movingCount(int threshold, int rows, int cols)
    {
        if (!rows || !cols) return 0;
        if (2*fun(0) > threshold) return 0;
        q.push({0,0});
        vis[0][0] = 1;
        bfs(threshold,rows,cols);
        return count;
    }
};