LeetCode 230. Kth Smallest Element in a BST

题目

Given a binary search tree, write a function kthSmallest to find the kth smallest element in it.

Note:

You may assume k is always valid, 1 ≤ k ≤ BST’s total elements.

思路

给定一个二叉搜索树，实现一个函数kthSmallest，找到这个BST中第k小的数。

根据BST的性质：

左子树中的所有节点都小于根节点，

右子树中的所有节点都大于根节点

可以想到利用中序遍历（左->中->右）来遍历BST后得到的序列是一个递增的序列，然后返回这个序列中第k个数即可。

用递归实现中序遍历

需要注意的是，递归函数第二个参数是待返回容器，函数中会修改这个值，所以要用引用来作为参数。

代码

/**
* 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:
int kthSmallest(TreeNode* root, int k) {
if(root == NULL)
return -1;
vector<int> res;
res = inorder(root,res);
return res[k-1];
}

vector<int> inorder(TreeNode* root,vector<int>& res)//引用调用
{
if(root == NULL)
return res;
if(root->left != NULL)
res=inorder(root->left,res);
res.push_back(root->val);
if(root->right != NULL)
res=inorder(root->right,res);
return res;
}
};