Kth Smallest Element in a BST

这道题不难。

但是卡了一会，原因是if(count=k-1) 我没写==。 然后突然想起吉林大学的一位C++老师，碰到 这种情况，我们要写k-1 ==count！！！

递归思路如下

      1.若左子树有k-1个节点，返回根的值

      2.若左子树有多余k-1个，则在左子树找第k小的

      3.若左子树少于k-1个，则在右子树找 k-左子树节点数-1小的

代码如下：class Solution {
public:
   
    int kthSmallest(TreeNode* root, int k) {
        if(!root)
            return 0;
        int count = count_subtree_nodes(root->left);
        //cout<<count<<endl;
        if(count==k-1)
            return root->val;
        else
            if(count>k-1)
                return kthSmallest(root->left,k);
            else
                return kthSmallest(root->right,k-1-count);
    }
    int count_subtree_nodes(TreeNode *root)
    {
        if(!root)
            return 0;
        return count_subtree_nodes(root->left)+count_subtree_nodes(root->right)+1;
    }
   
};