Same Tree

zoovely·2024년 7월 18일
0
post-thumbnail

💬 문제

[문제 링크]

두 이진 트리의 시작점 노드 p, q
두 트리가 정확히 같은 형태인지 (값 포함) true/false로 반환

✍️ 나의 풀이

/**
 * 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} p
 * @param {TreeNode} q
 * @return {boolean}
 */
var isSameTree = function(p, q) {
    let pVal = p ? p.val : null;
    let qVal = q ? q.val : null;

    if (pVal === null && qVal === null)
        return true;
    if (pVal !== qVal)
        return false;
    
    let left = isSameTree(p.left, q.left);
    let right = isSameTree(p.right, q.right);
    
    return left && right;
};

isSameTree 함수를 재귀함수로 구현
동시에 두 트리의 왼쪽 끝까지 갔다가 돌아오면서 오른쪽을 쭉 탐색하는 구조
탐색 중 node가 null이라면 leaf이므로
두 트리가 똑같이 null이라면 true를 반환하여 이전 단계로 돌아감
그 외에 한 쪽만 null이거나, 둘의 값이 다르다면
더 이상 탐색하지 않아도 되므로 false 반환

📌 결과

Accepted
Runtime 49ms (Beats 78.14%)
Memory 49.27MB (Beats 33.91%)

📚 러닝 포인트

우선은 DFS가 익숙해서 재귀 함수로 풀어보았다. 알고리즘은 같지만 표현을 다르게 해서 거의 세줄 정도로 끝낼 수 있지만, 보기 좀 더 편하게 코드를 풀어서 썼다. 최악의 경우 모든 노드를 탐색해야 하고, 가장 마지막 depth에 서로 다른 노드가 존재한다면 DFS를 사용하는게 빠르겠으나 그 외에는 BFS를 사용하는 것이 더 적절하다고 판단했다. 그래서 BFS로도 구현을 해봤다.

var isSameTree = function(p, q) {
    let queue = [[p, q]];
    
    while (queue[0]) {
        let [p, q] = queue.shift();

        if (p === null && q === null)
            continue;
        if ((p ? p.val : null) !== (q ? q.val : null))
            return false;

        queue.push([p.left, q.left]);
        queue.push([p.right, q.right]);
    }

    return true;
};

이전 문제와는 달리 depth를 구분할 필요가 없으므로 while 안에 for문이 없다. 조건문은 DFS와 비슷하게, 두 노드가 똑같이 null이라면 queue에 더 이상 넣지 않고 넘어가고, 두 노드가 다르다면 false를 바로 반환한다. 어쩌면 DFS와 달리 돌아오면서 연산하지 않고 바로 함수를 종료하기 때문에 더 빠른걸 수도 있겠다. 참고로 결과는 44ms / 48.8MB로 DFS보다 빨랐다. (리트 코드 특유의 테스트 차이일 수도 있지만)

profile
나도 할 수 있을까?

0개의 댓글