leetcode链接:https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/
给定一棵二叉搜索树,请找出其中第k大的节点。
示例 1:
输入: root = [3,1,4,null,2], k = 1
3
/ \
1 4
\
2
输出: 4示例 2:
输入: root = [5,3,6,2,4,null,null,1], k = 3
5
/ \
3 6
/ \
2 4
/
1
输出: 4
解题思路:由于是二叉树,因此可以采用递归,又根据二叉搜索树的性质,节点是有序的,最右下角的节点是最大值,找到第k大的节点,我们可以按照右根左的顺序进行dfs,设置一个辅助变量,记录遍历到第几个元素,当变量等于k时,该节点就是要寻找的节点。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int result,k;
public int kthLargest(TreeNode root, int k) {
this.k = k;
dfs(root);
return result;
}
public void dfs(TreeNode root){
if(root==null) return;
dfs(root.right);
if(k==0) return;
if(--k==0) result = root.val;
dfs(root.left);
}
}