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

2019-03-05  本文已影响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];
        
    }
};
上一篇下一篇

猜你喜欢

热点阅读