[LeetCode] Cut Off Trees for Golf Event 为高尔夫赛事砍树

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

You are asked to cut off trees in a forest for a golf event. The forest is represented as a non-negative 2D map, in this map:

0 represents the obstacle can't be reached.
1 represents the ground can be walked through.
The place with number bigger than 1 represents a tree can be walked through, and this positive number represents the tree's height.

You are asked to cut off all the trees in this forest in the order of tree's height - always cut off the tree with lowest height first. And after cutting, the original place has the tree will become a grass (value 1).

You will start from the point (0, 0) and you should output the minimum steps you need to walk to cut off all the trees. If you can't cut off all the trees, output -1 in that situation.

You are guaranteed that no two trees have the same height and there is at least one tree needs to be cut off.

Example 1:

Input:

[

 [1,2,3],

 [0,0,4],

 [7,6,5]

]

Output: 6

Example 2:

Input:

[

 [1,2,3],

 [0,0,0],

 [7,6,5]

]

Output: -1

Example 3:

Input:

[

 [2,3,4],

 [0,0,5],

 [8,7,6]

]

Output: 6

Explanation: You started from the point (0,0) and you can cut off the tree in (0,0) directly without walking.

Hint: size of the given matrix will not exceed 50x50.

这道题让我们砍掉所有高度大于1的树，而且要求是按顺序从低到高来砍，那么本质实际上还是要求任意两点之间的最短距离啊。对于这种类似迷宫遍历求最短路径的题，BFS是不二之选啊。那么这道题就对高度相邻的两棵树之间调用一个BFS，所以我们可以把BFS的内容放倒子函数helper中来使用。那么我们首先就要将所有的树从低到高进行排序，我们遍历原二位矩阵，将每棵树的高度及其横纵坐标取出来，组成一个三元组，然后放到vector中，之后用sort对数组进行排序，因为sort默认是以第一个数字排序，这也是为啥我们要把高度放在第一个位置。然后我们就遍历我们的trees数组，我们的起始位置是（0，0），终点位置是从trees数组中取出的树的位置，然后调用BFS的helper函数，这个BFS的子函数就是很基本的写法，没啥过多需要讲解的地方，会返回最短路径的值，如果无法到达目标点，就返回-1。所以我们先检查，如果helper函数返回-1了，那么我们就直接返回-1，否则就将cnt加到结果res中。然后更新我们的起始点为当前树的位置，然后循环取下一棵树即可，参见代码如下：

class Solution {

public:

    int cutOffTree(vector<vector<int>>& forest) {

        int m = forest.size(), n = forest[].size(), res = , row = , col = ;

        vector<vector<int>> trees;

        for (int i = ; i < m; ++i) {

            for (int j = ; j < n; ++j) {

                if (forest[i][j] > ) trees.push_back({forest[i][j], i, j});

            }

        }

        sort(trees.begin(), trees.end());

        for (int i = ; i < trees.size(); ++i) {

            int cnt = helper(forest, row, col, trees[i][], trees[i][]);

            if (cnt == -) return -;

            res += cnt;

            row = trees[i][];

            col = trees[i][];

        }

        return res;

    }

    int helper(vector<vector<int>>& forest, int row, int col, int treeRow, int treeCol) {

        if (row == treeRow && col == treeCol) return ;

        int m = forest.size(), n = forest[].size(), cnt = ;

        queue<int> q{{row * n + col}};

        vector<vector<int>> visited(m, vector<int>(n));

        vector<int> dir{-, , , , -};

        while (!q.empty()) {

            ++cnt;

            for (int i = q.size(); i > ; --i) {

                int r = q.front() / n, c = q.front() % n; q.pop();

                for (int k = ; k < ; ++k) {

                    int x = r + dir[k], y = c + dir[k + ];

                    if (x <  || x >= m || y <  || y >= n || visited[x][y] ==  || forest[x][y] == ) continue;

                    if (x == treeRow && y == treeCol) return cnt;

                    visited[x][y] = ;

                    q.push(x * n + y);

                }

            }

        }

        return -;

    }

};