标签:node cnt TreeNode val Offer int 54 二叉 return
由于是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);
}
}
标签:node,cnt,TreeNode,val,Offer,int,54,二叉,return 来源: https://www.cnblogs.com/nullpointer-c/p/15857720.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。