开启我的LeetCode刷题日记:675. 为高尔夫比赛砍树

115 阅读3分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第19天,点击查看活动详情

编程世界总是离不了算法

最近在看框架源码时,会有很多算法的实现逻辑,有时候会感到吃力

于是决定蹭着假期,加强算法和数据结构相关的知识

那怎么提升呢?

其实我知道算法这东西没有捷径,多写多练才能提升,于是我开启我的LeetCode刷题之旅

第一阶段目标是:200道,每天12

为了不乱,本系列文章目录分为三部分:

  1. 今日题目:xxx
  2. 我的思路
  3. 代码实现

今日题目: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] <= 109

来源:力扣(LeetCode) 链接:leetcode.cn/problems/cu… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

我的思路

早上的时候思路好乱,一直想dp,把自己弄的特别乱;晚上想了下,从低到高对所有树排个序,从 0,0 出发,寻找相邻两点的最短路径即可

  • 把有树的位置,都扔到数组里,按树的高度从小到大排序
  • 提供一个前驱点 pre,查找相邻两点的最短步数 BFS 就行
  • 如果找不到返回-1
  • 总距离等于距离相近树的距离的和

代码实现

class Solution {
    public int cutOffTree(List<List<Integer>> forest) {
        List<int[]> treePoses = new ArrayList<>();
        int m = forest.size();
        int n = forest.get(0).size();
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(forest.get(i).get(j)>1){
                    treePoses.add(new int[]{i,j});
                }
            }
        }

        treePoses.sort(new Comparator<int[]>() {
            @Override
            public int compare(int[] o1, int[] o2) {
                return forest.get(o1[0]).get(o1[1]) - forest.get(o2[0]).get(o2[1]);
            }
        });

        int[] pre = new int[2];
        int ans = 0;
        for(int i = 0; i < treePoses.size(); i++){
            int dis = getDis(pre,treePoses.get(i),forest);
            if(dis == -1) return -1;
            ans += dis;
            pre = treePoses.get(i);
        }
        return ans;
    }

    private int getDis(int[] from, int[] to,List<List<Integer>> forest){
        if(from[0]==to[0] && from[1]==to[1]) return 0;
        int m = forest.size();
        int n = forest.get(0).size();
        int[][] dp = new int[m][n];
        for(int i = 0; i < m; i++){
            Arrays.fill(dp[i],Integer.MAX_VALUE);
        }

        int v = forest.get(from[0]).get(from[1]);
        int tov = forest.get(to[0]).get(to[1]);
        dp[from[0]][from[1]] = 0;
        Queue<int[]> queue = new LinkedList<>();
        queue.offer(from);
        int[] direction = new int[]{0,-1,0,1,0};
        while (!queue.isEmpty()){
            int[] pos = queue.poll();
            for(int i = 0; i < 4; i++){
                int x = pos[0]+direction[i];
                int y = pos[1]+direction[i+1];
                if(x>=0&&x<m&&y>=0&&y<n&&dp[x][y]>dp[pos[0]][pos[1]]+1&&forest.get(x).get(y)>0){
                    dp[x][y] = dp[pos[0]][pos[1]]+1;
                    if(x==to[0]&&y==to[1]) return dp[x][y];
                    queue.offer(new int[]{x,y});
                }
            }
        }
        return -1;
    }
}

总结

实现方式其实有很多,这里仅供参考~

由于刚开始刷题,也不知道从哪里刷好,如果前辈们有好的建议,希望不吝赐教,感谢🌹