剑指offer系列36----二叉搜索树的第k个节点

【题目】给定一颗二叉搜索树，请找出其中的第k大的结点。
* 例如， 5
* /
* 3 7
* / /
* 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。 中序遍历：2 3 4 5 6 7 8

package com.exe7.offer;

import java.util.Stack;

/**
 * 【题目】给定一颗二叉搜索树，请找出其中的第k大的结点。
 *            例如， 5 
 *            /  
 *           3   7 
 *          /  /  
 *          2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。 中序遍历：2 3 4 5 6 7 8
 * @author WGS
 *
 */
public class BiTreeKthNode {
    static class TreeNode{
        int val;
        TreeNode left=null;
        TreeNode right=null;
        public TreeNode(int val){
            this.val=val;
        }
    }
    
    public TreeNode getKthNode(TreeNode pRoot,int k){
        if(pRoot==null ) return pRoot;
        TreeNode curNode=pRoot;
        Stack<TreeNode> stack=new Stack<>();
        int p=0;
        
        while(curNode!=null || !stack.isEmpty()){
            while(curNode!=null){
                stack.push(curNode);
                curNode=curNode.left;
            }
            if(!stack.isEmpty()){
                curNode=stack.pop();
                p++;
                if(p==k) return curNode;
                curNode=curNode.right;
            }
        }
        return null;        
    }

    public static void main(String[] args) {
        
        BiTreeKthNode b=new BiTreeKthNode();
        TreeNode root=new TreeNode(5);
        TreeNode node1=new TreeNode(3);
        TreeNode node2=new TreeNode(7);
        TreeNode node3=new TreeNode(2);
        TreeNode node4=new TreeNode(4);
        TreeNode node5=new TreeNode(6);
        TreeNode node6=new TreeNode(8);
        root.left=node1;
        root.right=node2;
        node1.left=node3;
        node1.right=node4;
        node2.left=node5;
        node2.right=node6;
        
        TreeNode node=b.getKthNode(root, 2);
        System.out.println(node.val);
    }

}