Leetcode - Kth Smallest Element in BST

[分析]
思路1: 递归中序遍历,返回中序遍历中的第k个数。
思路2: 迭代方式实现的中序遍历,遍历到第k个数即返回。
思路3: 如果可以修改数结构,增加一个leftCount属性,记录当前节点左子树节点个数。

[ref]
[url]http://bookshadow.com/weblog/2015/07/02/leetcode-kth-smallest-element-bst/[/url]


public class Solution {
// Method 1
public int kthSmallest1(TreeNode root, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
inorder(root, list);
return list.get(k - 1);
}
public void inorder(TreeNode root, ArrayList<Integer> list) {
if (root == null) return;
inorder(root.left, list);
list.add(root.val);
inorder(root.right, list);
}

// Method 2
public int kthSmallest(TreeNode root, int k) {
LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
TreeNode curr = root;
int n = 0;
while (curr != null || !stack.isEmpty()) {
if (curr != null) {
stack.push(curr);
curr = curr.left;
} else {
curr = stack.pop();
n++;
if (n == k)
return curr.val;
// if (curr.right != null) // leads to timeout
curr = curr.right;
}
}
return 0;
}

// Method 3
public int kthSmallest(TreeNode root, int k) {
while (root != null) {
if (root.leftCount == k - 1)
return root.val;
else if (root.leftCount >= k)
root = root.left;
else {
root = root.right;
k -= root.leftCount + 1;
}
}
return 0;
}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值