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

题目描述

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

[java]  view plain  copy

1. /* 
2. public class TreeNode { 
3.     int val = 0; 
4.     TreeNode left = null; 
5.     TreeNode right = null; 
6.  
7.     public TreeNode(int val) { 
8.         this.val = val; 
9.  
10.     } 
11.  
12. } 
13. */  
14. import java.util.Stack;  
15. public class Solution {  
16.     TreeNode KthNode(TreeNode pRoot, int k)  
17.     {  
18.         if(pRoot==null) return null;  
19.         int i = 0;        
20.         Stack<TreeNode> stack = new Stack<TreeNode>();        
21.         TreeNode p = pRoot;  
22.         while(p!=null || stack.size()!=0){                
23.             while(p!=null){               
24.                 stack.add(p);  
25.                 p = p.left;  
26.             }  
27.               
28.             if(stack.size()!=0){                  
29.                 p = stack.pop();  
30.                 if(++i==k){return p;}                 
31.                 p = p.right;  
32.             }  
33.         }  
34.         return null;  
35.     }  
36.   
37.   
38. }  

原文链接http://blog.csdn.net/crazy__chen/article/details/45113461