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];
}
};