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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

class Solution
{
private:
    int kthNode;

    void iterateTree(TreeNode *i_root, int &m_k)
    {
        if(i_root->right) iterateTree(i_root->right, m_k);

        m_k--;
        if(m_k == 0)
        {
            kthNode = i_root->val;
            return ;
        }

        if(i_root->left) iterateTree(i_root->left, m_k);
    }
public:
    // int kthLargest(TreeNode* root, int k)
    int kthLargest(TreeNode *i_root, int k)
    {
        iterateTree(i_root, k);

        return kthNode;
    }
};