1. 程式人生 > >劍指offer----二叉搜尋樹的第K個節點----java實現

劍指offer----二叉搜尋樹的第K個節點----java實現

思路:

根據二叉搜尋樹的特點來解決,非空二叉搜尋樹是左子樹上的所有節點比根節點小,右子樹上的所有節點比根節點大,其左右子樹同樣遵循這個規則。

所以二叉搜尋樹的中序遍歷結果,就是一個按照從大到小的順序排序的序列,訪問第K個節點時就終止程式即可。

TreeNode KthNode(TreeNode pRoot, int k)
	    {
	        if(pRoot == null || k <= 0)
	            return null;
	        TreeNode current = pRoot;
	        TreeNode kNode = null;//用來儲存第K個節點
	        int count = 0;
	        LinkedList<TreeNode> stack = new LinkedList<TreeNode>();
	        //中序遍歷二叉搜尋樹
	        while(!stack.isEmpty() || current != null)
	       {
	          while(current!=null)
	         {
	        	stack.push(current);
	        	current = current.left;
	         }
	          if(!stack.isEmpty())
	          {
	        	  current = stack.pop();//每出棧一個節點讓count++;對比count和K的值,第k個節點時跳出迴圈返回節點,
	        	  count++;
	        	  if(count == k)
	        	  {
	        		  kNode = current;
	        		  break;
	        	  }
	        	  current = current.right;
	          }
	       }
	        return kNode;
	    }