450. Delete Node in a BST

Given a root node reference of a BST and a key, delete the node with the given key in the BST. Return the root node reference (possibly updated) of the BST.
Basically, the deletion can be divided into two stages:
Search for a node to remove.
If the node is found, delete the node.
Note: Time complexity should be O(height of tree).

Solution:

因為是BST,所以先利用二分查找到要刪除的node第队,若不是BST則需要依次找(Solution2)哮塞,找到node后的要刪除的幾種情況:
node doesn't have left or right - return null
node only has left subtree- return the left subtree
node only has right subtree- return the right subtree

  • node has both left and right - find the minimum value in the right subtree, 就是右樹的最左下node_lb,將其替換要刪除的node(替換連接)凳谦,并按照node_lb only has right subtree的方式刪除node_lb
    Time Complexity: O(N) Space Complexity: O(1) (worst case O(n) for 遞歸緩存)


    IMG_0143.JPG

Solution1 Code:

class Solution {
    private TreeNode deleteRootNode(TreeNode root) {
        if (root == null) {
            return null;
        }
        if (root.left == null) {
            return root.right;
        }
        if (root.right == null) {
            return root.left;
        }
        TreeNode next = root.right;
        TreeNode pre = null;
        for(; next.left != null; pre = next, next = next.left);
        next.left = root.left;
        if(root.right != next) {
            pre.left = next.right;
            next.right = root.right;
        }
        return next;
    }
    
    public TreeNode deleteNode(TreeNode root, int key) {
        TreeNode cur = root;
        TreeNode pre = null;
        while(cur != null && cur.val != key) {
            pre = cur;
            if (key < cur.val) {
                cur = cur.left;
            } else if (key > cur.val) {
                cur = cur.right;
            }
        }
        if (pre == null) {
            return deleteRootNode(cur);
        }
        if (pre.left == cur) {
            pre.left = deleteRootNode(cur);
        } else {
            pre.right = deleteRootNode(cur);
        }
        return root;
    }
}

**Solution2 Code: **

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if (!root) return nullptr;
        if (root->val == key) {
            if (!root->right) {
                TreeNode* left = root->left;
                delete root;
                return left;
            }
            else {
                TreeNode* right = root->right;
                while (right->left)
                    right = right->left;
                swap(root->val, right->val);    
            }
        }
        root->left = deleteNode(root->left, key);
        root->right = deleteNode(root->right, key);
        return root;
    }
};
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末忆畅,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子尸执,更是在濱河造成了極大的恐慌家凯,老刑警劉巖,帶你破解...
    沈念sama閱讀 206,968評論 6 482
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件如失,死亡現(xiàn)場離奇詭異绊诲,居然都是意外死亡,警方通過查閱死者的電腦和手機褪贵,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,601評論 2 382
  • 文/潘曉璐 我一進店門驯镊,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人竭鞍,你說我怎么就攤上這事板惑。” “怎么了偎快?”我有些...
    開封第一講書人閱讀 153,220評論 0 344
  • 文/不壞的土叔 我叫張陵冯乘,是天一觀的道長。 經(jīng)常有香客問我晒夹,道長裆馒,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 55,416評論 1 279
  • 正文 為了忘掉前任丐怯,我火速辦了婚禮喷好,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘读跷。我一直安慰自己梗搅,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 64,425評論 5 374
  • 文/花漫 我一把揭開白布效览。 她就那樣靜靜地躺著无切,像睡著了一般。 火紅的嫁衣襯著肌膚如雪丐枉。 梳的紋絲不亂的頭發(fā)上哆键,一...
    開封第一講書人閱讀 49,144評論 1 285
  • 那天,我揣著相機與錄音瘦锹,去河邊找鬼籍嘹。 笑死闪盔,一個胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的辱士。 我是一名探鬼主播锭沟,決...
    沈念sama閱讀 38,432評論 3 401
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼识补!你這毒婦竟也來了族淮?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 37,088評論 0 261
  • 序言:老撾萬榮一對情侶失蹤凭涂,失蹤者是張志新(化名)和其女友劉穎祝辣,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體切油,經(jīng)...
    沈念sama閱讀 43,586評論 1 300
  • 正文 獨居荒郊野嶺守林人離奇死亡蝙斜,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,028評論 2 325
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了澎胡。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片孕荠。...
    茶點故事閱讀 38,137評論 1 334
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖攻谁,靈堂內(nèi)的尸體忽然破棺而出稚伍,到底是詐尸還是另有隱情,我是刑警寧澤戚宦,帶...
    沈念sama閱讀 33,783評論 4 324
  • 正文 年R本政府宣布个曙,位于F島的核電站,受9級特大地震影響受楼,放射性物質(zhì)發(fā)生泄漏垦搬。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,343評論 3 307
  • 文/蒙蒙 一艳汽、第九天 我趴在偏房一處隱蔽的房頂上張望猴贰。 院中可真熱鬧,春花似錦河狐、人聲如沸米绕。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,333評論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽义郑。三九已至蝶柿,卻和暖如春丈钙,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背交汤。 一陣腳步聲響...
    開封第一講書人閱讀 31,559評論 1 262
  • 我被黑心中介騙來泰國打工雏赦, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留劫笙,地道東北人。 一個月前我還...
    沈念sama閱讀 45,595評論 2 355
  • 正文 我出身青樓星岗,卻偏偏與公主長得像填大,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子俏橘,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 42,901評論 2 345

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