LeetCode 226 - Invert Binary Tree (Easy)

문제

LeetCode - 226번

풀이 과정

주어진 이진 트리를 뒤집는 연산을 수행하는 문제입니다.

다른분들의 풀이법을 보니 level order traversal 을 사용하신 분들도 있었지만 저는 dfs 를 사용했습니다.

각 서브트리의 루트 노드를 기준으로 왼쪽 서브트리와 오른쪽 서브트리를 invert 해준 다음
해당 서브트리의 루트노드를 반환해줍니다.

이러한 재귀 호출을 반복하면 최종적으로 기존의 root 는 invert 된 트리가 됩니다.

코드

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var invertTree = function (root) {
  function invert(root) {
    if (!root) return null;

    const left = invert(root.left);
    const right = invert(root.right);

    [root.left, root.right] = [right, left];

    return root;
  }

  invert(root);

  return root;
};