剑指offer-二叉搜索树的第k个结点

题目描述

给定一颗二叉搜索树，请找出其中的第k小的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

分析

通过中序遍历列出节点顺序列表

更加优化的解决办法是直接在遍历的时候将k值带进去O(k)-O(n)

import java.util.ArrayList;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;

public TreeNode(int val) {
this.val = val;

}

}
*/
public class Solution {
TreeNode KthNode(TreeNode pRoot, int k) {
if (pRoot == null) {
return null;
}
ArrayList<TreeNode> list = new ArrayList<TreeNode>();
inOrderTree(pRoot, list);
if (list.size() < k || k == 0) {
return null;
} else {
return list.get(k-1);
}
}

private void inOrderTree(TreeNode pRoot, ArrayList<TreeNode> list) {
if(pRoot == null) {
return;
}
inOrderTree(pRoot.left, list);
list.add(pRoot);
inOrderTree(pRoot.right, list);
}
}

O(k):

int cnt = 0;

TreeNode KthNode(TreeNode pRoot, int k) {
if (pRoot == null)
return null;
TreeNode t = KthNode(pRoot.left, k);
if (t != null)
return t;
if (++cnt < k)
return KthNode(pRoot.right, k);
else if (cnt == k)
return pRoot;
return null;
}