LeetCode剑指 Offer 13. 机器人的运动范围（DFS + 回溯）

最新推荐文章于 2024-05-14 20:06:41 发布

GSX_M

最新推荐文章于 2024-05-14 20:06:41 发布

阅读量545

点赞数 1

分类专栏：力扣刷题文章标签： leetcode 算法 DFS

本文链接：https://blog.csdn.net/m0_52169086/article/details/123640915

版权

力扣刷题专栏收录该内容

64 篇文章 0 订阅

订阅专栏

力扣

解题思路:

开辟一个相同规格的二维数组 book，将不符合条件的位置设置为 1 ，再利用DFS深度优先搜索，找出能到达的最大格子。

class Solution {
public:
    int dir[4][2] = {1 , 0 , -1 , 0 , 0 , 1 , 0 , -1};
    void DFS(vector<vector<int>>& book ,int row ,int col ,int x ,int y ,int& count)
    {
        book[x][y] = 1; //符合条件设置为已使用
        count++;
        
        for(int i = 0 ; i < 4 ;++i)//四个方位
        {
            int newx = x + dir[i][0];
            int newy = y + dir[i][1];

            if(newx < 0 || newx >= row || newy < 0 || newy >= col || book[newx][newy] != 0)
            continue; //不符合条件继续找

            DFS(book , row ,col ,newx ,newy ,count);
        }
    }
   
    int movingCount(int m, int n, int k) 
    {
        if(k == 0)
        return 1;

        vector<vector<int>> book(m , vector<int>(n , 0));

        for(int i = 0 ; i < m ;++i) //初始化将不符合条件的位置设置为 1,
        {
            for(int j = 0 ; j < n ;++j)
            {
                int ti = i;
                int tj = j;
                int sum = 0;

                while(ti)
                {
                    sum += ti % 10;
                    ti /= 10;
                }

                while(tj)
                {
                    sum += tj % 10;
                    tj /= 10;
                }

                if(sum > k)
                {
                    book[i][j] = 1;
                }
            }
        }
        
        int count = 0; 
        DFS(book , m ,n ,0 , 0 ,count);

        return count;
    }
};