competitive-programming2022년 9월 6일1분 분량

[Leetcode] 814. Binary Tree Pruning 풀이

수정일: 2022년 9월 6일

Trees

지원 언어:enko

#Problem

814. Binary Tree Pruning

#Approach

각 트리 노드에 대해 아래 과정을 반복합니다.

왼쪽 자식을 가지치기(prune)합니다.
오른쪽 자식을 가지치기(prune)합니다.
왼쪽과 오른쪽 자식이 null이고 이 노드가 값 1을 가지고 있지 않으면, null을 반환합니다 (가지치기).

#Code

/**
 * author: jooncco
 * written: 2022. 9. 6. Tue. 13:14:14 [UTC+9]
 **/

class Solution {
    private final int TARGET_VALUE = 1;

    public TreeNode pruneTree(TreeNode root) {
        if (root == null) return null;

        root.left= pruneTree(root.left);
        root.right= pruneTree(root.right);
        if (root.left == null && root.right == null && root.val != TARGET_VALUE) return null;
        return root;
    }
}

#Complexity

Time: $O(n)$
Space: $O(n)$