2022-05-23每日一题

675. 为高尔夫比赛砍树

你被请来给一个要举办高尔夫比赛的树林砍树。树林由一个 m x n 的矩阵表示， 在这个矩阵中：

0 表示障碍，无法触碰
1 表示地面，可以行走
比 1 大的数 表示有树的单元格，可以行走，数值表示树的高度
每一步，你都可以向上、下、左、右四个方向之一移动一个单位，如果你站的地方有一棵树，那么你可以决定是否要砍倒它。

你需要按照树的高度从低向高砍掉所有的树，每砍过一颗树，该单元格的值变为 1（即变为地面）。

你将从 (0, 0) 点开始工作，返回你砍完所有树需要走的最小步数。 如果你无法砍完所有的树，返回 -1 。

可以保证的是，没有两棵树的高度是相同的，并且你至少需要砍倒一棵树。

示例 1：

输入：forest = [[1,2,3],[0,0,4],[7,6,5]]
输出：6
解释：沿着上面的路径，你可以用 6 步，按从最矮到最高的顺序砍掉这些树。
示例 2：

输入：forest = [[1,2,3],[0,0,0],[7,6,5]]
输出：-1
解释：由于中间一行被障碍阻塞，无法访问最下面一行中的树。
示例 3：

输入：forest = [[2,3,4],[0,0,5],[8,7,6]]
输出：6
解释：可以按与示例 1 相同的路径来砍掉所有的树。
(0,0) 位置的树，可以直接砍去，不用算步数。

提示：

m == forest.length
n == forest[i].length
1 <= m, n <= 50
0 <= forest[i][j] <= 10⁹

思路：题目大意是从起点(0,0)开始，按数值从小到大依次把树砍掉，求完成整个过程需要移动的最小步数，一般最小步数问题，可以用到BFS，可以通过排序，每次求一个点到另一个点的最小步数，把它们求和，就是最终结果，比如示例一，会先找起点到数值为2的点的最小步数，然后再找数值为2的点到数值为3的点的最小步数，一直寻找下去，之后把它们的步数都加起来就能达到整体最小，注意大于1的树会被加入排序集合，遇到0时会被阻碍

class Solution {
    int n,m;
    List<int[]> list = new ArrayList<>();
    public int cutOffTree(List<List<Integer>> forest) {
        n = forest.size();
        m = forest.get(0).size();
        for(int i = 0;i < n;i++){
            for(int j=0;j<m;j++){
                if(forest.get(i).get(j)>1){
                    list.add(new int[]{forest.get(i).get(j),i,j});
                }
            }
        }
        Collections.sort(list,(a,b)->a[0]-b[0]);//对要砍的数高度排序
        if(forest.get(0).get(0) == 0) //起点就是障碍，返回-1
            return -1;
        int x = 0,y = 0,ans = 0;
        for(int[] li:list){
            int nx = li[1],ny = li[2];
            int step = bfs(x,y,nx,ny,forest);//两点之间的最小步数
            if(step==-1)
                return -1;
            ans += step;
            x = nx;
            y = ny;
        }
        return ans;
    }
    int[][] dirs = new int[][]{{0,1},{0,-1},{-1,0},{1,0}};
    int bfs(int x,int y,int tx,int ty,List<List<Integer>> forest){
        if(x==tx && y==ty)
            return 0;
        boolean[][] vis = new boolean[n][m];
        Queue<int[]> q = new LinkedList<>();
        q.add(new int[]{x,y});
        vis[x][y] = true;
        int ans = 0;
        while(!q.isEmpty()){
            int size = q.size();
            while(size-->0){
                int[] info = q.poll();
                int xx = info[0];
                int yy = info[1];
                for(int[] di:dirs){
                    int nx = xx+di[0];
                    int ny = yy+di[1];
                    if(nx < 0 || nx >= n || ny < 0 || ny >= m || forest.get(nx).get(ny) == 0 ||   vis[nx][ny]){
                        continue;
                    }
                    if(nx == tx && ny == ty){
                        return ans+1;
                    }
                    q.add(new int[]{nx,ny});
                    vis[nx][ny] = true;
                }
            }
            ans++;
        }
        return -1;
    }
}