题目描述
给定一棵二叉搜索树,请找出其中的第k小的TreeNode结点。
示例1
输入
{5,3,7,2,4,6,8},3
返回值
{4}
说明
按结点数值大小顺序第三小结点的值为4
代码:
import java.util.*;
public class Solution {
TreeNode KthNode(TreeNode pRoot, int k) {
if(k <= 0){
return null;
}
Stack<TreeNode> st = new Stack<>();
while (pRoot != null || !st.empty()) {
if (pRoot != null) {
st.add(pRoot);
pRoot = pRoot.left;
} else {
pRoot = st.pop();
k--;
if(k == 0){
return pRoot;
}
pRoot = pRoot.right;
}
}
return null;
}
}
思路概述:
二叉搜索树的特性是中序遍历是从小到大排列,所以我们基于中序遍历,每从栈中弹出节点时,让K减一。当K为0时,指针指向节点即为第K小的结点。