LeetCode Kth Smallest Element in a 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 countNode(TreeNode* root)//计算子树有多少个节点，其实还可以将这个函数直接融到下面的函数中，用K来标记功能即可。
    {
        if(root==NULL)    return 0;
        int L=countNode(root->left);
        int R=countNode(root->right);
        return L+R+1;
    }
    int kthSmallest(TreeNode* root, int k) 
    {
        int cnt=countNode(root->left);
        if(cnt+1==k)    return root->val;
        if(cnt>=k)    kthSmallest(root->left,k);
        else    kthSmallest(root->right,k-cnt-1);    
    }
};