두 이진 트리의 시작점 노드 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보다 빨랐다. (리트 코드 특유의 테스트 차이일 수도 있지만)