剑指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);
     }

 }