[LeetCode 99] Recover Binary Search Tree

Two elements of a binary search tree (BST) are swapped by mistake.

Recover the tree without changing its structure.

Example 1:

Input: [1,3,null,null,2]

   1
  /
 3
  \
   2

Output: [3,1,null,null,2]

   3
  /
 1
  \
   2

Example 2:

Input: [3,1,4,null,null,2]

  3
 / \
1   4
   /
  2

Output: [2,1,4,null,null,3]

  2
 / \
1   4
   /
  3

Follow up:

  • A solution using O(n) space is pretty straight forward.
  • Could you devise a constant space solution?

Solution: InOrder Traverse

  1. 用中序遍歷 (正常時遍歷的值為升序)
  2. 但是有2個節(jié)點出現(xiàn)的錯誤间唉,就打破了升序绞灼。
    • 那么只要找到第一個 current.val < prev.val,那么prev就是第一個出錯的節(jié)點
    • 然后再找到第二個 current.val < prev.val呈野,那么prev就是第二個出錯的節(jié)點
  3. 交換即可
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public void recoverTree(TreeNode root) {
        TreeNode first = null;
        TreeNode second = null;
        TreeNode prev = null;
        
        Stack<TreeNode> tracker = new Stack<> ();
        TreeNode current = root;
        
        while (!tracker.isEmpty () || current != null) {
            if (current != null) {
                tracker.push (current);
                current = current.left;
            } else {
                current = tracker.pop ();
                
                // find the two nodes which have been swapped incorrectly
                // cannot use if else here, for example [3,1,4,null,null,2], which in-order is 1 3 2 4, prev: 3, current 2. 
                // the first and second nodes are found at the same time. so we cannot use if else.
                if (prev != null && current.val < prev.val) {
                    if (first == null)
                        first = prev;
                    second = current;
                }
                
                prev = current;
                current = current.right;
            }
        }
        
        int temp = first.val;
        first.val = second.val;
        second.val = temp;
        
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末低矮,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子被冒,更是在濱河造成了極大的恐慌军掂,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,386評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件昨悼,死亡現(xiàn)場離奇詭異蝗锥,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)率触,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,142評論 3 394
  • 文/潘曉璐 我一進(jìn)店門终议,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人葱蝗,你說我怎么就攤上這事穴张。” “怎么了两曼?”我有些...
    開封第一講書人閱讀 164,704評論 0 353
  • 文/不壞的土叔 我叫張陵皂甘,是天一觀的道長。 經(jīng)常有香客問我合愈,道長叮贩,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,702評論 1 294
  • 正文 為了忘掉前任佛析,我火速辦了婚禮益老,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘寸莫。我一直安慰自己捺萌,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 67,716評論 6 392
  • 文/花漫 我一把揭開白布膘茎。 她就那樣靜靜地躺著桃纯,像睡著了一般酷誓。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上态坦,一...
    開封第一講書人閱讀 51,573評論 1 305
  • 那天盐数,我揣著相機(jī)與錄音,去河邊找鬼伞梯。 笑死玫氢,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的谜诫。 我是一名探鬼主播漾峡,決...
    沈念sama閱讀 40,314評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼喻旷!你這毒婦竟也來了生逸?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,230評論 0 276
  • 序言:老撾萬榮一對情侶失蹤且预,失蹤者是張志新(化名)和其女友劉穎槽袄,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體辣之,經(jīng)...
    沈念sama閱讀 45,680評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡掰伸,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,873評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了怀估。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 39,991評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡合搅,死狀恐怖多搀,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情灾部,我是刑警寧澤康铭,帶...
    沈念sama閱讀 35,706評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站赌髓,受9級特大地震影響从藤,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜锁蠕,卻給世界環(huán)境...
    茶點故事閱讀 41,329評論 3 330
  • 文/蒙蒙 一夷野、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧荣倾,春花似錦悯搔、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,910評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽通危。三九已至,卻和暖如春灌曙,著一層夾襖步出監(jiān)牢的瞬間菊碟,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,038評論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留醉旦,地道東北人六水。 一個月前我還...
    沈念sama閱讀 48,158評論 3 370
  • 正文 我出身青樓,卻偏偏與公主長得像忍燥,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子隙姿,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 44,941評論 2 355

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