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

题目描述

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

解法：

对于二叉搜索树而已，中序遍历就按照节点值从小到大遍历，因此进行中序遍历，第k个节点为所求。
 

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


    public TreeNode(int val) {
        this.val = val;


    }


}
*/
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>();
        TreeNode cur = pRoot;
        int index = 0;   
		while(cur!=null || !stack.isEmpty()){
			while(cur!=null){				
				stack.push(cur);
				cur = cur.left;
			}
			TreeNode treeNode = stack.pop();
			if(++index==k){
                return treeNode;
            }
			if(treeNode!=null){
				cur = treeNode.right;
			}
		}		
        return null;
    }
}