剑指Offer - 二叉搜索树的第k个结点(Java实现)

题目描述：

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

二叉搜索树：
二叉查找树（Binary Search Tree），（又：二叉搜索树，二叉排序树）它或者是一棵空树，或者是具有下列性质的二叉树： 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值； 它的左、右子树也分别为二叉排序树。

思路分析：
在二叉搜索树中，根据节点数值的大小找出第k个小的节点；所以本题想到的是先排序（根据节点大小数值进行排序），然后在找出第k为上的节点即可；二叉搜索树的中序遍历的结果是递增的数列，所以本体的核心在于中序遍历。
方法一：
采用递归的方式先获得中序遍历的结果，然后在从结果中找出第k个节点
代码实现如下：

import java.util.ArrayList;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}
*/
//二叉搜索树的中序遍历的结果为递增数列
//先求得中序遍历的结果
public class Solution {
    public TreeNode KthNode(TreeNode pRoot, int k)
    {
        
        ArrayList<TreeNode> list = new ArrayList<>();
        inOrderRecur(pRoot,list);
        int len = list.size();
        if(k < 1 || k > len){
            return null;
        } 
        TreeNode result = list.get(k-1);
        return result;
    }
    public void inOrderRecur(TreeNode pRoot,ArrayList<TreeNode> list){
        if(pRoot == null){
            return;
        }
        inOrderRecur(pRoot.left,list);
        list.add(pRoot);
        inOrderRecur(pRoot.right,list);
    }

}

方法二：
采用方法一固然可以，但是时间复杂度过高，只能先获得中序遍历的结果才能找出位于第k个位置的节点，但是若采用非递归的方式，则可以在出栈的时候记录这个节点出栈的位次是否和要求的位置一致，若一致则直接返回该节点即可，非递归的代码实现如下：

import java.util.Stack;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}
*/
//非递归的方式实现中序遍历
public class Solution {
    public TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k < 1){
            return null;
        }
        int count = 0;
        TreeNode cur = pRoot;
        Stack<TreeNode> stack = new Stack<>();
        while(cur != null || !stack.isEmpty()){
            while( cur != null){
                stack.push(cur);
                cur = cur.left;
            }
            cur = stack.pop();
            count++;
            if(count == k){
                return cur;
            }
            cur = cur.right;
        }
        return null;
    }
}