美文网首页
leetcode 230. 二叉搜索树中第K小的元素

leetcode 230. 二叉搜索树中第K小的元素

作者: 橘子煲汤 | 来源:发表于2019-03-05 19:32 被阅读0次

    这道题使用到了中序遍历 首先使用中序遍历 搜索整棵树 然后把节点放入vector中 二叉搜索树经过中序遍历后得到的数组就是一个排序后的数组 然后return k-1就可以了 当然 也可以不适用额外的空间 但是我觉得使用额外的空间的话 程序更清晰一些 所以 还是用数组保存下遍历后的节点值吧

    /**
     * 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:
        vector<int> res;
        void kth(TreeNode * root)
        {
            if(root!=nullptr){
                kth(root->left);
                res.push_back(root->val);
                kth(root->right);
                
            }
        }
        int kthSmallest(TreeNode* root, int k) {
            kth(root);
            return res[k-1];
            
        }
    };
    

    相关文章

      网友评论

          本文标题:leetcode 230. 二叉搜索树中第K小的元素

          本文链接:https://www.haomeiwen.com/subject/vnzouqtx.html