劍指offer----二叉搜尋樹的第K個節點----java實現
阿新 • • 發佈:2019-01-07
思路:
根據二叉搜尋樹的特點來解決,非空二叉搜尋樹是左子樹上的所有節點比根節點小,右子樹上的所有節點比根節點大,其左右子樹同樣遵循這個規則。
所以二叉搜尋樹的中序遍歷結果,就是一個按照從大到小的順序排序的序列,訪問第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; }