2024.3.17力扣每日一题——最小高度树

2024-04-05 05:28

本文主要是介绍2024.3.17力扣每日一题——最小高度树,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

2024.3.17

      • 题目来源
      • 我的题解
        • 方法一 深度优先遍历
        • 方法二 广度优先遍历
        • 方法三 拓扑排序

题目来源

力扣每日一题;题序:310

我的题解

方法一 深度优先遍历

从每一个节点开始进行深度优先遍历并计算以该节点为根节点的树的深度,使用哈希表存储对应深度有哪些节点,最后取出key最小对应的value的作为结果。超时。

时间复杂度:O( n 2 n^2 n2)
空间复杂度:O(n)

TreeMap<Integer,Set<Integer>> map=new TreeMap<>();
public List<Integer> findMinHeightTrees(int n, int[][] edges) {List<Integer>[] g=createTree(n,edges);for(int i=0;i<n;i++){int minDeep=dfs(g,i,i,-1,1);System.out.println(minDeep);Set<Integer> l=map.getOrDefault(minDeep,new HashSet<>());l.add(i);map.put(minDeep,l);}return new ArrayList<>(map.firstEntry().getValue());
}
public List<Integer>[] createTree(int n,int[][] edges){List<Integer>[] g=new ArrayList[n];for(int i=0;i<n;i++){g[i]=new ArrayList<>();}for(int[] t:edges){int from = t[0];int to = t[1];g[from].add(to);g[to].add(from);}return g;
}
public int  dfs(List<Integer>[] g,int root,int cur,int pre,int deepth){int minDeep=0;for(int next:g[cur]){if(next!=pre){int t=dfs(g,root,next,cur,deepth+1);minDeep=Math.max(minDeep,t);}}return minDeep+1;
}

由于一定是树,所以具有最小深度的根节点只有可能最多两个节点作为个根时的深度是相同的。所以,可以求该图中最长的直径,中间节点就是需要求的根节点。所以,不用对所有节点都进行深度优先遍历,只需要遍历两个节点就行了。
时间复杂度:O(n)
空间复杂度:O(n)

//优化版本
TreeMap<Integer,Set<Integer>> map=new TreeMap<>();
List<Integer> path=new ArrayList<>();
int deep=0;
public List<Integer> findMinHeightTrees(int n, int[][] edges) {List<Integer>[] g=createTree(n,edges);//优化的地方List<Integer> p=new ArrayList<>();int x=0;p.add(x);dfs(g,x,-1,1,p);p=new ArrayList<>();int y=path.get(path.size()-1);p.add(y);dfs(g,y,-1,1,p);int sz=path.size();if(sz%2==0){return path.subList(sz/2-1,sz/2+1);}else{return path.subList(sz/2,sz/2+1);}
}
public List<Integer>[] createTree(int n,int[][] edges){List<Integer>[] g=new ArrayList[n];for(int i=0;i<n;i++){g[i]=new ArrayList<>();}for(int[] t:edges){int from = t[0];int to = t[1];g[from].add(to);g[to].add(from);}return g;
}
public void  dfs(List<Integer>[] g,int cur,int pre,int deepth,List<Integer> p){if(deepth>deep){path=new ArrayList<>(p);deep=deepth;}for(int next:g[cur]) {if (next != pre) {p.add(next);dfs(g, next, cur, deepth + 1, p);p.remove(p.size() - 1);}}
}
方法二 广度优先遍历

在方法一的优化版本基础上,也可以采用广度优先遍历来求最长直径。

时间复杂度:O(n)
空间复杂度:O(n)

TreeMap<Integer,Set<Integer>> map=new TreeMap<>();
List<Integer> path=new ArrayList<>();
int deep=0;
public List<Integer> findMinHeightTrees(int n, int[][] edges) {List<Integer>[] g=createTree(n,edges);//变化之处在于采用广度优先遍历求最长直径int x=0;bfs(g,x);int y=path.get(path.size()-1);bfs(g,y);int sz=path.size();if(sz%2==0){return path.subList(sz/2-1,sz/2+1);}else{return path.subList(sz/2,sz/2+1);}
}
public List<Integer>[] createTree(int n,int[][] edges){List<Integer>[] g=new ArrayList[n];for(int i=0;i<n;i++){g[i]=new ArrayList<>();}for(int[] t:edges){int from = t[0];int to = t[1];g[from].add(to);g[to].add(from);}return g;
}
public void bfs(List<Integer>[] g,int cur){Queue<Integer> queue=new LinkedList<>();queue.offer(cur);int[] parent=new int[g.length];//记录路径parent[cur]=-1;boolean[] visited=new boolean[g.length];visited[cur]=true;//最后一层的节点int end=cur;while(!queue.isEmpty()){int sz=queue.size();for (int i = 0; i < sz; i++) {int t=queue.poll();boolean f=false;for(int next:g[t]){if(visited[next])continue;parent[next]=t;visited[next]=true;queue.offer(next);f=true;}if(!f)end=t;}}path.clear();while(end!=-1){path.add(0,end);end=parent[end];}
}
方法三 拓扑排序

最小深度也可以看做是一直删除叶子节点,最后剩下的一个或者两个节点就是满足条件的节点。这种每次删除叶子节点的操作实质就是拓扑排序。

时间复杂度:O(n)
空间复杂度:O(n)

public List<Integer> findMinHeightTrees(int n, int[][] edges) {//注意:使用拓扑排序需要将只有一个节点的情况单独考虑if(n==1)return Arrays.asList(0);int[] tp=new int[n];List<Integer>[] g=createTree(n,edges,tp);Queue<Integer> queue=new LinkedList<>();for(int i=0;i<n;i++){if(tp[i]==1)queue.offer(i);}return tpFunc(g,tp,n,queue);
}
public List<Integer>[] createTree(int n,int[][] edges,int[] tp){List<Integer>[] g=new ArrayList[n];for(int i=0;i<n;i++){g[i]=new ArrayList<>();}for(int[] t:edges){int from = t[0];int to = t[1];g[from].add(to);g[to].add(from);tp[from]++;tp[to]++;}return g;
}
public List<Integer> tpFunc(List<Integer>[] g,int[] tp,int remain,Queue<Integer> queue){List<Integer> res=new ArrayList<>();while(remain>2){int sz=queue.size();remain-=sz;for(int i=0;i<sz;i++){int t=queue.poll();for(int next:g[t]){tp[t]--;tp[next]--;if(tp[next]==1)queue.offer(next);}}}while(!queue.isEmpty()){res.add(queue.poll());}return res;
}

有任何问题,欢迎评论区交流,欢迎评论区提供其它解题思路(代码),也可以点个赞支持一下作者哈😄~

这篇关于2024.3.17力扣每日一题——最小高度树的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/877715

相关文章

使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)

《使用Python从PPT文档中提取图片和图片信息(如坐标、宽度和高度等)》PPT是一种高效的信息展示工具,广泛应用于教育、商务和设计等多个领域,PPT文档中常常包含丰富的图片内容,这些图片不仅提升了... 目录一、引言二、环境与工具三、python 提取PPT背景图片3.1 提取幻灯片背景图片3.2 提取

java获取图片的大小、宽度、高度方式

《java获取图片的大小、宽度、高度方式》文章介绍了如何将File对象转换为MultipartFile对象的过程,并分享了个人经验,希望能为读者提供参考... 目China编程录Java获取图片的大小、宽度、高度File对象(该对象里面是图片)MultipartFile对象(该对象里面是图片)总结java获取图片

vue基于ElementUI动态设置表格高度的3种方法

《vue基于ElementUI动态设置表格高度的3种方法》ElementUI+vue动态设置表格高度的几种方法,抛砖引玉,还有其它方法动态设置表格高度,大家可以开动脑筋... 方法一、css + js的形式这个方法需要在表格外层设置一个div,原理是将表格的高度设置成外层div的高度,所以外层的div需要

poj 1258 Agri-Net(最小生成树模板代码)

感觉用这题来当模板更适合。 题意就是给你邻接矩阵求最小生成树啦。~ prim代码:效率很高。172k...0ms。 #include<stdio.h>#include<algorithm>using namespace std;const int MaxN = 101;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int n

poj 1287 Networking(prim or kruscal最小生成树)

题意给你点与点间距离,求最小生成树。 注意点是,两点之间可能有不同的路,输入的时候选择最小的,和之前有道最短路WA的题目类似。 prim代码: #include<stdio.h>const int MaxN = 51;const int INF = 0x3f3f3f3f;int g[MaxN][MaxN];int P;int prim(){bool vis[MaxN];

poj 2349 Arctic Network uva 10369(prim or kruscal最小生成树)

题目很麻烦,因为不熟悉最小生成树的算法调试了好久。 感觉网上的题目解释都没说得很清楚,不适合新手。自己写一个。 题意:给你点的坐标,然后两点间可以有两种方式来通信:第一种是卫星通信,第二种是无线电通信。 卫星通信:任何两个有卫星频道的点间都可以直接建立连接,与点间的距离无关; 无线电通信:两个点之间的距离不能超过D,无线电收发器的功率越大,D越大,越昂贵。 计算无线电收发器D

poj 1734 (floyd求最小环并打印路径)

题意: 求图中的一个最小环,并打印路径。 解析: ans 保存最小环长度。 一直wa,最后终于找到原因,inf开太大爆掉了。。。 虽然0x3f3f3f3f用memset好用,但是还是有局限性。 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#incl

hdu 1102 uva 10397(最小生成树prim)

hdu 1102: 题意: 给一个邻接矩阵,给一些村庄间已经修的路,问最小生成树。 解析: 把已经修的路的权值改为0,套个prim()。 注意prim 最外层循坏为n-1。 代码: #include <iostream>#include <cstdio>#include <cstdlib>#include <algorithm>#include <cstri

poj 2175 最小费用最大流TLE

题意: 一条街上有n个大楼,坐标为xi,yi,bi个人在里面工作。 然后防空洞的坐标为pj,qj,可以容纳cj个人。 从大楼i中的人到防空洞j去避难所需的时间为 abs(xi - pi) + (yi - qi) + 1。 现在设计了一个避难计划,指定从大楼i到防空洞j避难的人数 eij。 判断如果按照原计划进行,所有人避难所用的时间总和是不是最小的。 若是,输出“OPETIMAL",若

poj 2135 有流量限制的最小费用最大流

题意: 农场里有n块地,其中约翰的家在1号地,二n号地有个很大的仓库。 农场有M条道路(双向),道路i连接着ai号地和bi号地,长度为ci。 约翰希望按照从家里出发,经过若干块地后到达仓库,然后再返回家中的顺序带朋友参观。 如果要求往返不能经过同一条路两次,求参观路线总长度的最小值。 解析: 如果只考虑去或者回的情况,问题只不过是无向图中两点之间的最短路问题。 但是现在要去要回