剑指 Offer 54. 二叉搜索树的第k大节点
2022/1/30 23:07:46
本文主要是介绍剑指 Offer 54. 二叉搜索树的第k大节点,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
剑指 Offer 54. 二叉搜索树的第k大节点
由于是BST,那么我们很容易想到将中序遍历的结果取第k大的数即可。
所以我们先用一个list把中序遍历得到的结果存储起来,再从中取第k大的那个即可。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { List<Integer> path = new ArrayList<>(); public int kthLargest(TreeNode root, int k) { traverse(root); return path.get(path.size() - k); } private void traverse(TreeNode node) { if(node == null) { return ; } traverse(node.left); path.add(node.val); traverse(node.right); } }
上述算法的时间复杂度为\(O(lgn)\),n个节点个数,空间复杂度为\(O(n)\),我们可以发现,其实很多节点是不用存储的,并且list新建和插入比较耗时,于是考虑如何降低空间,我们用一个cnt
表示当前是遍历到第cnt
个,每遍历一个cnt对应相加,直到cnt = k
,返回ans即可。
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { int ans, cnt; public int kthLargest(TreeNode root, int k) { dfs(root, k); return ans; } private void dfs(TreeNode node, int k) { if(node == null) return ; dfs(node.right, k); cnt++; if(cnt == k) { ans = node.val; return ; } dfs(node.left, k); } }
这篇关于剑指 Offer 54. 二叉搜索树的第k大节点的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-07-08如何在敏捷项目中实现高效测试?
- 2024-07-08用户故事一定要有 “So that...” 吗?
- 2024-07-04TiDB 资源管控的对撞测试以及最佳实践架构
- 2024-07-03万字长文聊聊Web3的组成架构
- 2024-07-02springboot项目无法注册到nacos-icode9专业技术文章分享
- 2024-06-26结对编程到底难不难?答案在这里
- 2024-06-19《2023版Java工程师》课程升级公告
- 2024-06-15matplotlib作图不显示3D图,怎么办?
- 2024-06-1503-Loki 日志监控
- 2024-06-1504-让LLM理解知识 -Prompt