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

题目描述

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

 

中序遍历二叉搜索树，找到第k个结点，除了使用递归，还可以使用栈实现非递归写法

public class Solution {//树 my
    int index = 0;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null){
            return null;
        }
        return getK(pRoot,k);
    }
    TreeNode getK(TreeNode root,int k){
        if(root==null){
            return null;
        }
        TreeNode left = getK(root.left,k);
        if(left!=null){
            return left;
        }
        index++;
        if(index==k){
            return root;
        }
        TreeNode right = getK(root.right,k);
        if(right!=null){
            return right;
        }
        return null;
    }
}

代码可简化为

public class Solution {
    int index = 0;
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot!=null){
            TreeNode left = KthNode(pRoot.left,k);
            if(left!=null){
                return left;
            }
            index++;
            if(index==k){
                return pRoot;
            }
            TreeNode right = KthNode(pRoot.right,k);
            if(right!=null){
                return right;
            }
        }
        return null;
    }
}