LeetCode-173-二叉搜尋樹迭代器
阿新 • • 發佈:2022-04-02
二叉搜尋樹迭代器
題目描述:實現一個二叉搜尋樹迭代器類BSTIterator ,表示一個按中序遍歷二叉搜尋樹(BST)的迭代器:
- BSTIterator(TreeNode root) 初始化 BSTIterator 類的一個物件。BST 的根節點 root 會作為建構函式的一部分給出。指標應初始化為一個不存在於 BST 中的數字,且該數字小於 BST 中的任何元素。
- boolean hasNext() 如果向指標右側遍歷存在數字,則返回 true ;否則返回 false 。
- int next()將指標向右移動,然後返回指標處的數字。
注意,指標初始化為一個不存在於 BST 中的數字,所以對 next() 的首次呼叫將返回 BST 中的最小元素。
你可以假設 next() 呼叫總是有效的,也就是說,當呼叫 next() 時,BST 的中序遍歷中至少存在一個下一個數字。
示例說明請見LeetCode官網。
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/binary-search-tree-iterator/
著作權歸領釦網路所有。商業轉載請聯絡官方授權,非商業轉載請註明出處。
解法一:中序遍歷
首先,在BSTIterator內宣告一個List為values用來存二叉樹的節點值,在構造方法內通過中序遍歷的方式將節點值初始化到values中。
然後,宣告一個index標識當前位置,
next()
方法和hasNext()
方法的實現即依賴於index和values進行判斷即可。
import com.kaesar.leetcode.TreeNode; import java.util.ArrayList; import java.util.List; public class LeetCode_173 { public static void main(String[] args) { // 測試用例 TreeNode root = new TreeNode(7); root.left = new TreeNode(3); root.right = new TreeNode(15); root.right.left = new TreeNode(9); root.right.right = new TreeNode(20); BSTIterator bSTIterator = new BSTIterator(root); System.out.println(bSTIterator.next()); // 返回 3 System.out.println(bSTIterator.next()); // 返回 7 System.out.println(bSTIterator.hasNext()); // 返回 True System.out.println(bSTIterator.next()); // 返回 9 System.out.println(bSTIterator.hasNext()); // 返回 True System.out.println(bSTIterator.next()); // 返回 15 System.out.println(bSTIterator.hasNext()); // 返回 True System.out.println(bSTIterator.next()); // 返回 20 System.out.println(bSTIterator.hasNext()); // 返回 False } } class BSTIterator { private int index; private List<Integer> values; public BSTIterator(TreeNode root) { this.index = 0; this.values = new ArrayList<>(); inOrder(root, values); } /** * 中序遍歷 * * @param root * @param values */ private void inOrder(TreeNode root, List<Integer> values) { if (root == null) { return; } inOrder(root.left, values); values.add(root.val); inOrder(root.right, values); } public int next() { return values.get(index++); } public boolean hasNext() { return index < values.size(); } }
【每日寄語】 中流砥柱,力挽狂瀾,具天才,立大業,拯斯民於衽席,奠國運如磐石,非大英雄無以任之。