BFS解决最短路问题_为高尔夫比赛砍树_C++

---

1. 题目解析+算法分析

---

leetcode链接：https://leetcode.cn/problems/cut-off-trees-for-golf-event/submissions/

1. 大白话讲解

• 在一个森林里砍树（矩阵），只能从低到高砍树，并且0表示障碍物，除0之外的其他地方都是可以走动的。起始位置在[0][0]处，问要走多少步才能将树砍完。

2. 示例

• 砍到第一个树1，需要0步；
• 接着要砍第二棵树2，要从1的位置走两步；
• 砍3时只需要从2的位置走1步；
• 以此类推直到将所有树砍完，计算一共的步数。

3. 算法分析

• 第一步：先将所有树的位置记录下来，并从低到高进行排序；
• 第二步：按照顺序开始砍树，用bfs计算每一次从起始位置走到目标位置需要几步。如果走不到直接返回-1。

---

2. 代码实现

---

class Solution {
    int dx[4] = {0, 0, 1, -1};
    int dy[4] = {1, -1, 0, 0};
    bool vis[51][51];
    int m, n;
    
public:
    void init()
    {
        for (int i = 0; i < 51; i++)
            for (int j = 0; j < 51; j++)
                vis[i][j] = false;
    }

    int bfs(vector<vector<int>>& forest, int bx, int by, int ex, int ey)
    {
        if (bx == ex && by == ey) return 0;     // 处理边界情况，如果起始位置和终止位置一样了，还遍历个锤子

        init();         // 初始化vis数组
        queue<pair<int, int>> q;
        q.push({bx, by});
        vis[bx][by] = true;

        int step = 0;
        while(q.size())
        {
            step++;
            int sz = q.size();
            while (sz--)
            {
                auto [a, b] = q.front();
                q.pop();
                for (int i = 0; i < 4; i++)
                {
                    int x = a + dx[i], y = b + dy[i];
                    if (x >= 0 && x < m && y >= 0 && y < n && forest[x][y] != 0 && !vis[x][y])
                    {
                        if (x == ex && y == ey) return step;
                        q.push({x, y});
                        vis[x][y] = true;
                    }
                }
            }
        }
        return -1;
    }

    int cutOffTree(vector<vector<int>>& forest) 
    {
        m = forest.size(), n = forest[0].size();
        // 1. 准备工作：找出砍树的顺序
        vector<pair<int, int>>trees;
        for (int i = 0; i < m; i++)
            for (int j = 0; j < n; j++)
            if (forest[i][j] > 1) trees.push_back({i, j});

        sort(trees.begin(), trees.end(), [&](const pair<int, int>& p1, const pair<int, int>& p2)
        {
            return forest[p1.first][p1.second] < forest[p2.first][p2.second];
        });

        // 2. 按照顺序砍树
        int bx = 0, by = 0; // 起始坐标
        int ret = 0;
        for (auto& [a, b] : trees)
        {
            int step = bfs(forest, bx, by, a, b);
            if (step == -1) return -1;
            ret += step;
            bx = a, by = b;
        }
        return ret;
    }
};

---