《树》剑指 Offer 54. 二叉搜索树的第k大节点《leetcode》

剑指 Offer 54. 二叉搜索树的第k大节点

难度简单90

给定一棵二叉搜索树，请找出其中第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

 

限制：

1 ≤ k ≤ 二叉搜索树元素个数

思路分析：

二叉搜索树的中序遍历即为从小到大排列，第k大的反向中序即可，即右根左，

然后优化，加一个counter，只需要到第k个返回即可。 时间复杂度，O（k）

AC代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */


class Solution {
    public int counter=0;
    public int result=0;
    public void rMid(TreeNode root,int k)
    {   
        if(root!=null)
        {   
            rMid(root.right,k);
            counter++;
            if(counter==k)
            {    
                result=root.val;
                return;
            }
            rMid(root.left,k);
        }
        return;
    }

    public int kthLargest(TreeNode root, int k) {
        if(root==null)
            return 0;
        counter=0;
        result=0;
        rMid(root,k);
        return result;
    }
}

再提一点：

有想了解二叉搜索树的，可以访问这篇博客。