题目描述
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
参考思路
二叉搜索树的特点是左结点的值都比根结点的值小(存在左结点的前提下),右结点的值都比根结点的值大(存在右结点的前提下)。题目需要找出第K小的结点,相当于对二叉搜索树进行中序遍历。
参考代码
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
int count = 0;
TreeNode resNode;
TreeNode KthNode(TreeNode pRoot, int k)
{
if(k <= 0 || pRoot == null){
return null;
}
DFS(pRoot,k);
return resNode;
}
//深度优先遍历,里面递归调用时也需要按照"左根右"的顺序执行
void DFS(TreeNode root,int k){
if(count < k && root.left != null){
DFS(root.left,k);
}
//这边表示每次访问跟累加并跟K进行比较
if(++count == k){
resNode = root;
}
if(count < k && root.right != null){
DFS(root.right,k);
}
}
}