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

最新推荐文章于 2022-09-09 22:43:00 发布

Amberfd

最新推荐文章于 2022-09-09 22:43:00 发布

阅读量129

点赞数

分类专栏：剑指offer Leetcode 文章标签： c++

本文链接：https://blog.csdn.net/Amberfd/article/details/115670509

版权

剑指offer 同时被 2 个专栏收录

72 篇文章 0 订阅

订阅专栏

Leetcode

20 篇文章 0 订阅

订阅专栏

在这里插入图片描述

解法一

广度优先搜索，使用pair，以及队列queue，先进先出

//广度优先搜索
class Solution {
    int get(int x){
        int res = 0;
        for ( ; x; x/=10)
        {
            res+=x % 10;
        }
        return res;
    }
public:
    int movingCount(int m, int n, int k) {
        if(!k) return 1;
        queue<pair<int,int>> Q; //先进先出
        //向左和向下数组
        int dx[2] = {1,0};
        int dy[2] = {0,1};
        vector<vector<int>> vis(m,vector<int>(n,0));
        Q.push(make_pair(0,0));
        vis[0][0] = 1;
        int ans =1 ;
        while(!Q.empty()){
            auto [x,y] = Q.front();
            Q.pop();
            for (int i = 0; i < 2; i++)
            {
                int tx = dx[i]+x;
                int ty = dy[i]+y;
                if(tx<0 || tx>=m || ty<0 || ty>=n|| vis[tx][ty]|| get(tx)+get(ty)>k) continue;
                Q.push(make_pair(tx,ty));
                vis[tx][ty] = 1;
                ans++;
            }
        }
        return ans;
    }
};

解法二

递推，找到上一个可达的格子，推断下一个是否可达，然后叠加ans

class Solution {
    int get(int x){
        int res = 0;
        for ( ; x; x = x/10)
        {
            res += x%10;
        }
        return res;
    }
 
public:
    int movingCount(int m, int n, int k) {
        if(!k) return 1;
        int ans = 1;
        vector<vector<int>> vis(m,vector<int>(n,0));
        vis[0][0]=1;
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if((i == 0 && j ==0) || get(i)+get(j)>k) continue;
                if(i-1>=0) vis[i][j] = vis[i][j] || vis[i-1][j];
                if(j-1>=0) vis[i][j] = vis[i][j] || vis[i][j-1];
                ans+=vis[i][j];
            }
        }
        return ans;
    }
};