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

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

题目描述

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

其实这道题问的就是二叉树的中序遍历
因为二叉树的中序遍历就是左、根、右
得到的排序就是一个从小到大的有序队列
因此这道题就是中序遍历的变形，只需加一个计数器，用来记录遍历到第几个节点。

import java.util.Stack;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k <= 0){
            return null;
        }
        Stack<TreeNode> stack = new Stack<>(); //建立栈
        TreeNode cur = pRoot;
        while(!stack.isEmpty()||cur!=null)
        {
            if(cur!=null)
            {
                stack.push(cur);//当前节点不为null，应该寻找左儿子
                cur = cur.left;//左儿子
            }
            else{
                cur = stack.pop();//当前节点null则弹出栈内元素，相当于按顺序输出最小值。
                if(--k==0)
                    return cur;
                cur = cur.right;
                }
        }
        return null;
   }
}
        

思路如下：
（先判断非法访问）
1.先设置一个栈，用于存放所经过的节点的信息
2.第一次经过的点不访问，而是先将该结点入栈
3.先遍历左子树，结束后，第二次访问该节点，就将该节点出栈，将该节点设为访问过的节点
4.当栈为空时就结束