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.知识点:
递归,二叉排序树