剑指offer第62题“二叉搜索树的第k个结点”

最后第五题

Posted by Cfeng on April 6, 2019

二叉搜索树的第k个结点

题目描述

给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。 ***

思路

当然可以把二叉搜索树每个点都遍历一遍,然后存在数组中,再排个序,取出第k个。 但熟悉概念的话可以知道,二叉搜索树的中序遍历就是从小到大遍历,所以当中序遍历到第k个时,即可返回。

Java实现

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    private int index = 0;
    private TreeNode ans = null;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(ans != null){
            return ans;
        }
        if(pRoot==null || k <= 0){
            return null;
        }
        if(pRoot.left!=null){
            ans = KthNode(pRoot.left, k);
        }
        index++;
        if(index == k){
            ans = pRoot;
            return ans;
        }
         if(pRoot.right!=null){
            ans = KthNode(pRoot.right, k);
        }
        return ans;
    }
}