[leetcode_230]Kth Smallest Element in a BST

求二叉搜索树的第k个数,中序遍历生成二叉搜索树,存入数组中,然后输出第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
class Solution {
public:
    vector<int> list;
    void queryTree(TreeNode * node)
    {
        if (NULL == node)
        {
            return;
        }
        if (NULL != node->left)
        {
            queryTree(node->left);
        }
        list.push_back(node->val);
        if (NULL != node->right)
        {
            queryTree(node->right);
        }
    }

    int kthSmallest(TreeNode* root, int k) {
        list.clear();
        queryTree(root);
        return list[k-1];
    }
};
Licensed under CC BY-NC-SA 4.0