题意:
给一个二叉排序树(bst),找出其中第k小的结点
分析:
我们就要去找从小到大是什么样的顺序,发现正好就是从上到下从左到右回溯搜索的顺序,那么递归深搜即可。那么我们对结点计数是在哪里呢?
我们来看遍历的过程中:到达当前结点,向左搜索,返回,向右搜索,返回。我们发现当前结点被走了三次,但是,并不意味着这三个位置计数都可以。虽然次数都只是一次,但是计数的顺序有区别。我们对一个结点的计数应该在搜索左边之后,搜索右边之前。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
int k = 0;
int val = 0;
public int kthSmallest(TreeNode root, int k) {
this.k = k;
helper(root);
return val;
}
private void helper(TreeNode node){
if(node == null){ //叶子节点的下一个结点(不满足条件的结点)
return;
}
//到达当前结点
helper(node.left); //向下(左)搜素
//从下(左)返回当前结点
k--;
if(k == 0){
val = node.val;
}
helper(node.right); //向下(右)搜索
//从下(右)返回当前结点
}
}