Leetcode 173. Binary Search Tree Iterator

Implement an iterator over a binary search tree (BST). Your iterator will be initialized with the root node of a BST.

Calling next() will return the next smallest number in the BST.

Note: next() and hasNext() should run in average O(1) time and uses O(h) memory, where h is the height of the tree.

思路:BST的中序遍歷就是一個升序,滿足題目的迭代要求。考慮到要O(h)的空間復雜度错沃,用椄胄模可以保存當前最小node的父節(jié)點礼饱,每次最小值在棧頂夯膀,O(1)的時間復雜度。

public 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.isEmpty();
}

/** @return the next smallest number */
public int next() {
    TreeNode node = stack.pop();
    TreeNode dummy = node.right;
    while (dummy != null) {
        stack.push(dummy);
        dummy = dummy.left;
    }
    return node.val;
}
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末溺健,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子钮蛛,更是在濱河造成了極大的恐慌鞭缭,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,122評論 6 505
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件魏颓,死亡現(xiàn)場離奇詭異岭辣,居然都是意外死亡,警方通過查閱死者的電腦和手機甸饱,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,070評論 3 395
  • 文/潘曉璐 我一進店門沦童,熙熙樓的掌柜王于貴愁眉苦臉地迎上來仑濒,“玉大人,你說我怎么就攤上這事偷遗《胀” “怎么了?”我有些...
    開封第一講書人閱讀 164,491評論 0 354
  • 文/不壞的土叔 我叫張陵氏豌,是天一觀的道長矗烛。 經(jīng)常有香客問我,道長箩溃,這世上最難降的妖魔是什么瞭吃? 我笑而不...
    開封第一講書人閱讀 58,636評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮涣旨,結(jié)果婚禮上歪架,老公的妹妹穿的比我還像新娘。我一直安慰自己霹陡,他們只是感情好和蚪,可當我...
    茶點故事閱讀 67,676評論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著烹棉,像睡著了一般攒霹。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上浆洗,一...
    開封第一講書人閱讀 51,541評論 1 305
  • 那天催束,我揣著相機與錄音,去河邊找鬼伏社。 笑死抠刺,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的摘昌。 我是一名探鬼主播速妖,決...
    沈念sama閱讀 40,292評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼聪黎!你這毒婦竟也來了罕容?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,211評論 0 276
  • 序言:老撾萬榮一對情侶失蹤稿饰,失蹤者是張志新(化名)和其女友劉穎锦秒,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體湘纵,經(jīng)...
    沈念sama閱讀 45,655評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡脂崔,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,846評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了梧喷。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片砌左。...
    茶點故事閱讀 39,965評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡脖咐,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出汇歹,到底是詐尸還是另有隱情屁擅,我是刑警寧澤,帶...
    沈念sama閱讀 35,684評論 5 347
  • 正文 年R本政府宣布产弹,位于F島的核電站派歌,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏痰哨。R本人自食惡果不足惜胶果,卻給世界環(huán)境...
    茶點故事閱讀 41,295評論 3 329
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望斤斧。 院中可真熱鬧早抠,春花似錦、人聲如沸撬讽。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,894評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽游昼。三九已至甘苍,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間烘豌,已是汗流浹背载庭。 一陣腳步聲響...
    開封第一講書人閱讀 33,012評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留扇谣,地道東北人昧捷。 一個月前我還...
    沈念sama閱讀 48,126評論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像罐寨,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子序矩,可洞房花燭夜當晚...
    茶點故事閱讀 44,914評論 2 355

推薦閱讀更多精彩內(nèi)容