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

1.题目：
给定一个二叉搜索树，编写一个函数 kthSmallest 来查找其中第 k 个最小的元素。
说明：
你可以假设 k 总是有效的，1 ≤ k ≤ 二叉搜索树元素个数。

示例 :
输入: root = [5,3,6,2,4,null,null,1], k = 3
       5
      / \
     3   6
    / \
   2   4
  /
 1
输出: 3

2.代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int getNodeNum(struct TreeNode* root) {
    if(!root)
        return 0;
    return getNodeNum(root->left)+getNodeNum(root->right)+1;
}

//第k则左子树要有k-1个节点。若小与k-1个节点则在右边，否则在左边。
//往左走都一样，还是找第k个节点。
//往右走则是是找第k-num-1个节点，'1'是根节点。

int kthSmallest(struct TreeNode* root, int k) {
    if(!root)
        return -1;
    int leftNums=getNodeNum(root->left);
    if(leftNums==k-1)                	   //中
        return root->val;
    else if(leftNums>k-1)            	   //还在左边
        return kthSmallest(root->left, k);
    else                              	   //在右边
        return kthSmallest(root->right, k-leftNums-1);    
}

3.知识点：

递归，二叉排序树