24. 机器人的运动范围

 

 

地上有一个 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

 

class Solution {
public:


    /*
     *
     * 在这里我们要好好考虑一下一个问题：
     * 这里说的是数位之和 大于一个阈值，因此这里的话我们需要对每个数字都进行一下相加的操作；
     * */



    int get_single_sum(int x){
        int s=0;
        while (x ) {
            s+= x%10;
            x = x/10;
        }
        return s;
    }

    int get_sum(pair<int,int> p){

        return get_single_sum(p.first) +  get_single_sum(p.second);
    }

/*
 * 以上两个函数为计算求和的函数
 * */



    int movingCount(int threshold, int rows, int cols)
    {
        int result =0; // 在这里定义结果为0

        if (!rows || !cols) return 0; //判断一下边界条件；

        vector<vector<bool>> grid(rows,vector<bool>(cols));// 定义一个数组来判断一下有没有被访问过,每个位置只能被访问一次

        queue <pair<int,int>> que;

        que.push(make_pair(0,0)); // 在这里我们从左上角 0,0 点开始遍历;

        int dx[4] = {-1, 0,1,0}; // 定义四个方向
        int dy[4] = {0,1,0,-1};

        while (que.size()){

            auto q = que.front();
            que.pop();// 要记得能够弹出来

            if(get_sum(q) > threshold ||  grid[q.first][q.second] ==true) continue; // 如果要是大于threshold 或者大于边界值的话需要接着进行；

            result ++; // result +1;

            grid[q.first][q.second] = true;// 把数据赋值为 True

            for(int i =0; i<4 ;i++){

                int x = q.first + dx[i]; // 在这里开始遍历四个方向的数据
                int y = q.second + dy[i];

                if(x>=0 && x<rows && y>=0 && y<cols){ // 这里要首先进行一下边界条件的判断 ,很重要在这里；
                    que.push({x,y});  // 把数据存起来；

                }

            }
        }

        return result; // 返回结果；

    }
};


// 加油加油 Try to make yourself more excellent...