Leetcode 剑指 Offer 54. 二叉搜索树的第k大节点【二叉树的中序遍历】

问题描述

解题报告

由于二叉搜索的中序遍历可以得到一个有序序列，所以本题用中序遍历来实现。
中序遍历有 递归 和 非递归 两种实现方式。
如果采用 递归 的方式，则需要将树中所有节点遍历完了之后，取出有序序列的第 k 大元素；
如果采用 非递归 的方式，则只需遍历到第 k 个节点，即可跳出程序。

另外需要注意的一点是：中序遍历时【左-中-右】得到的是从小到大的有序序列；【右-中-左】得到的是从大到小的有序序列，而本题宜采用【右-中-左】的遍历方式。

实现代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
// 逆向中序遍历，得到从大到小的有序序列
// class Solution {
// public:
//     int kthLargest(TreeNode* root, int k) {
//         vector<int>ans;
//         inorder(root, ans);
//         // for(int i=0;i<ans.size();i++) cout<<ans[i]<<" ";
//         return ans[k-1];
//     }
//     void inorder(TreeNode* root, vector<int>&ans){
//         if(root!=NULL){
//             inorder(root->right, ans);
//             ans.push_back(root->val);
//             inorder(root->left, ans);
//         }
//     }
// };

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

// 作者：OrangeMan
// 链接：https://leetcode-cn.com/problems/er-cha-sou-suo-shu-de-di-kda-jie-dian-lcof/solution/cjian-dan-yi-dong-dai-ma-ji-hu-shuang-bai-by-orang/
// 来源：力扣（LeetCode）
// 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

参考资料

[1] Leetcode 剑指 Offer 54. 二叉搜索树的第k大节点
[2] 题解区：OrangeMan