【剑指Offer】面试题54. 二叉搜索树的第k大节点

题目

给定一棵二叉搜索树,请找出其中第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 ≤ 二叉搜索树元素个数

思路

逆中序遍历。

代码

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

class Solution {
public:
    int kthLargest(TreeNode* root, int k) {
        stack<TreeNode*> st;      
        while (root || !st.empty()) {           
            while (root) {
                st.push(root);
                root = root->right;                
            }
            TreeNode *node = st.top();
            st.pop();
            if (--k == 0) return node->val;
            root = node->left;
        }
        return -1;
    }
};

声明:该文章系转载,转载该文章的目的在于更广泛的传递信息,并不代表本网站赞同其观点,文章内容仅供参考。

本站是一个个人学习和交流平台,网站上部分文章为网站管理员和网友从相关媒体转载而来,并不用于任何商业目的,内容为作者个人观点, 并不代表本网站赞同其观点和对其真实性负责。

我们已经尽可能的对作者和来源进行了通告,但是可能由于能力有限或疏忽,导致作者和来源有误,亦可能您并不期望您的作品在我们的网站上发布。我们为这些问题向您致歉,如果您在我站上发现此类问题,请及时联系我们,我们将根据您的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。