二叉搜索樹迭代器 -- 兩種解法

1. 中序遍歷后將節(jié)點(diǎn)保存至鏈表:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class BSTIterator {

    public BSTIterator(TreeNode root) {
        midOrder(root);
    }
    
    /** @return the next smallest number */
    public int next() {
        if(list != null && list.size() > index){
            return list.get(index++).val;
        }else{
            return -1;
        }
    }
    
    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        return (list != null && list.size() > index);
    }
    
    private int index = 0;
    private List<TreeNode> list = new ArrayList<>();
    private void midOrder(TreeNode root){
        if(root != null){
            midOrder(root.left);
            list.add(root);
            midOrder(root.right);
        }
    }
}

/**
 * Your BSTIterator object will be instantiated and called as such:
 * BSTIterator obj = new BSTIterator(root);
 * int param_1 = obj.next();
 * boolean param_2 = obj.hasNext();
 */

2. 用棧來(lái)存儲(chǔ)左側(cè)節(jié)點(diǎn)整吆,實(shí)現(xiàn)中序遍歷

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class BSTIterator {

    Stack<TreeNode> stack = new Stack<>();

    /** 初始化迭代器 */
    public BSTIterator(TreeNode root) {
        // 添加元素
        while(root!=null){
            stack.push(root);
            root = root.left;
        }
    }

    /** @return whether we have a next smallest number */
    public boolean hasNext() {
        return !stack.empty();
    }

    /** @return the next smallest number */
    public int next() {
        // 獲取并彈出最小元素
        TreeNode node = stack.pop();
        int min  = node.val;
        node = node.right;
        // 添加元素被彈出結(jié)點(diǎn)的右子樹上的較小元素
        while(node!=null){
            stack.push(node);
            node = node.left;
        }
        // 返回最小元素
        return min;
    }
}

/**
 * Your BSTIterator object will be instantiated and called as such:
 * BSTIterator obj = new BSTIterator(root);
 * int param_1 = obj.next();
 * boolean param_2 = obj.hasNext();
 */
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末散址,一起剝皮案震驚了整個(gè)濱河市冒窍,隨后出現(xiàn)的幾起案子莫换,更是在濱河造成了極大的恐慌瘦馍,老刑警劉巖洞豁,帶你破解...
    沈念sama閱讀 219,366評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件灭忠,死亡現(xiàn)場(chǎng)離奇詭異跷乐,居然都是意外死亡肥败,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,521評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門愕提,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)馒稍,“玉大人,你說我怎么就攤上這事浅侨∨耍” “怎么了?”我有些...
    開封第一講書人閱讀 165,689評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵如输,是天一觀的道長(zhǎng)鼓黔。 經(jīng)常有香客問我,道長(zhǎng)不见,這世上最難降的妖魔是什么澳化? 我笑而不...
    開封第一講書人閱讀 58,925評(píng)論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮稳吮,結(jié)果婚禮上缎谷,老公的妹妹穿的比我還像新娘。我一直安慰自己灶似,他們只是感情好列林,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,942評(píng)論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著酪惭,像睡著了一般希痴。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上春感,一...
    開封第一講書人閱讀 51,727評(píng)論 1 305
  • 那天砌创,我揣著相機(jī)與錄音,去河邊找鬼鲫懒。 笑死纺铭,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的刀疙。 我是一名探鬼主播舶赔,決...
    沈念sama閱讀 40,447評(píng)論 3 420
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼谦秧!你這毒婦竟也來(lái)了竟纳?” 一聲冷哼從身側(cè)響起撵溃,我...
    開封第一講書人閱讀 39,349評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎锥累,沒想到半個(gè)月后缘挑,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,820評(píng)論 1 317
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡桶略,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,990評(píng)論 3 337
  • 正文 我和宋清朗相戀三年语淘,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片际歼。...
    茶點(diǎn)故事閱讀 40,127評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡惶翻,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出鹅心,到底是詐尸還是另有隱情吕粗,我是刑警寧澤,帶...
    沈念sama閱讀 35,812評(píng)論 5 346
  • 正文 年R本政府宣布旭愧,位于F島的核電站颅筋,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏输枯。R本人自食惡果不足惜议泵,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,471評(píng)論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望桃熄。 院中可真熱鬧先口,春花似錦、人聲如沸蜻拨。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,017評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)缎讼。三九已至,卻和暖如春坑匠,著一層夾襖步出監(jiān)牢的瞬間血崭,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,142評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工厘灼, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留夹纫,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,388評(píng)論 3 373
  • 正文 我出身青樓设凹,卻偏偏與公主長(zhǎng)得像舰讹,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子闪朱,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,066評(píng)論 2 355