leetcode oj java 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.

二。1、解决思路：

由于BST的左孩子节点的值都小于父节点，右孩子节点的值都大于父节点的特性，我们可以递归的去寻找K小的元素。

假设左子树的长度是length

如果 length + 1 = k 那么刚好是根节点的值

如果 length + 1 > k 那么结果在左子树中，还是寻找第k个。

如果
length + 1 < k 那么结果在右子树中，此时已经排除了左边的length+1 个，需要寻找k-(length+1)个。

附代码：

public static int kthSmallest(TreeNode root, int k) {
if (root == null) {
return 0;
}
int leftSize = count(root.left);
if (k == leftSize + 1) {
return root.val;
} else if (k < leftSize + 1) {
return kthSmallest(root.left, k);
} else {
return kthSmallest(root.right, k - leftSize - 1);
}
}

public static int count(TreeNode root) {
if (root == null) {
return 0;
}
return 1 + count(root.left) + count(root.right);
}

解决思路二：

中序遍历树，把结果存入list中，一旦list的长度大于K的时候就返回。然后取list的第k-1个值即可。

代码：

public static int kthSmallest(TreeNode root, int k) {
List<Integer> re = new ArrayList<Integer>();
bianli(root, re, k);
return re.get(k - 1);
}

public static List<Integer> bianli(TreeNode root, List<Integer> re, int k) {
if (root == null || re.size() > k) {
return re;
}
bianli(root.left, re, k);
re.add(root.val);
bianli(root.right, re, k);
return re;
}