LeetCode-173-二叉搜尋樹迭代器

雄獅虎豹發表於2022-04-08
LeetCode

二叉搜尋樹迭代器

題目描述：實現一個二叉搜尋樹迭代器類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/probl...
著作權歸領釦網路所有。商業轉載請聯絡官方授權，非商業轉載請註明出處。

解法一：中序遍歷

首先，在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();
    }
}
【每日寄語】 中流砥柱，力挽狂瀾，具天才，立大業，拯斯民於衽席，奠國運如磐石，非大英雄無以任之。

相關文章