[LeetCode] 450. Delete Node in a BST, Medium

2025. 4. 10. 18:55·CodingTest/LeetCode
반응형

1. 문제

  • BST와 정수 key가 주어질 때, 해당 key를 제거한 BST를 반환하라.

2. 해결

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function deleteNode(root: TreeNode | null, key: number): TreeNode | null {
    if(!root) return null;
    
    if(root.val === key) {
        let childCount = 0;
    
        if(root.left) childCount++;
        if(root.right) childCount++;

        if(childCount === 0) {
            return null;
        } else if(childCount === 1) {
            return root.left || root.right;
        } else if(childCount ===2) {
            
            let successor = root.right;

            while(successor.left) {
                successor = successor.left;
            }
            root.val = successor.val;
            root.right = deleteNode(root.right, successor.val)
            return root;
        }
    }

    if(root.val > key)root.left = deleteNode(root.left, key);
    if(root.val < key)root.right = deleteNode(root.right,key)

    return root;
};
  • key와 일치하는 경우, 3가지 경우의 수를 나눠서 처리
    • 삭제 대상 자식이 없는 경우,
    • 삭제 대상 자식이 1개인 경우,
    • 삭제 대상 자식이 2개인 경우,
  • 여기서 2개인 경우는, 오른쪽 서브 트리의 최소값을 찾아 삭제 대상과 교체해줘야 BST가 깨지지 않는다.
저작자표시 (새창열림)
'CodingTest/LeetCode' 카테고리의 다른 글
  • [LeetCode] 235. Lowest Common Ancestor of a Binary Search Tree, Medium
  • [LeetCode] 703. Kth Largest Element in a Stream, Easy
  • [LeetCode] 701. Insert into a Binary Search Tree, Medium
  • [LeetCode] 700. Search in a Binary Search Tree, Easy
뜸부깅
뜸부깅
코딩에 대한 여러 개인적인 생각을 정리하고 공부를 하는 공간입니다!!
  • 뜸부깅
    코오오딩
    뜸부깅
  • 전체
    오늘
    어제
    • Note (429)
      • Skill (31)
        • Java & Spring (9)
        • Javascript & HTML & CSS (0)
        • React (0)
        • Next.js (22)
      • CodingTest (389)
        • 백준 온라인 저지(BOJ) (140)
        • 프로그래머스(Programmers) (79)
        • LeetCode (170)
      • Algorithm & Data Structure (6)
      • [Project] 포트폴리오 (3)
        • Front end (3)
        • Back end (0)
  • 블로그 메뉴

    • 홈
    • 태그
    • 방명록
  • 링크

  • 공지사항

  • 인기 글

  • 태그

    백준1427
    medium
    leetcode 2236
    백준7576자바
    Easy
    next 14
    Java
    TypeScript
    boj2108
    boj1427
    BOJ
    백준2751
    프로그래머스
    백준7576
    component-scan
    알고리즘
    백준
    자바
    백준1260
    meidum
  • 최근 댓글

  • 최근 글

  • hELLO· Designed By정상우.v4.10.3
뜸부깅
[LeetCode] 450. Delete Node in a BST, Medium
상단으로

티스토리툴바