【js】Leetcode每日一題-葉子相似的樹

kernel_developer發表於2021-05-10
JSLeetCode每日一題

【js】Leetcode每日一題-葉子相似的樹

【題目描述】

請考慮一棵二叉樹上所有的葉子，這些葉子的值按從左到右的順序排列形成一個 葉值序列 。

舉個例子，如上圖所示，給定一棵葉值序列為 (6, 7, 4, 9, 8) 的樹。

如果有兩棵二叉樹的葉值序列是相同，那麼我們就認為它們是 葉相似 的。

如果給定的兩個根結點分別為 root1 和 root2 的樹是葉相似的，則返回 true；否則返回 false 。

示例 1：

輸入：root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
輸出：true

示例 2：

輸入：root1 = [1], root2 = [1]
輸出：true

示例 3：

輸入：root1 = [1], root2 = [2]
輸出：false

示例 4：

輸入：root1 = [1,2], root2 = [2,2]
輸出：true

示例 5：

輸入：root1 = [1,2,3], root2 = [1,3,2]
輸出：false

提示：

給定的兩棵樹可能會有 1 到 200 個結點。
給定的兩棵樹上的值介於 0 到 200 之間。

【分析】

dfs

dfs收集完葉子到陣列中，然後比對兩個陣列是否完全相同。

時間複雜度：\(O(m+n)\)

/**
 * 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} root1
 * @param {TreeNode} root2
 * @return {boolean}
 */
var leafSimilar = function(root1, root2) {
    const dfs = function(root, seq){
        if(!root) return;
        if(!root.right&&!root.left) seq.push(root.val);
        if(root.left) dfs(root.left, seq);
        if(root.right) dfs(root.right, seq);
    }
    const seq1 = [];
    const seq2 = [];
    dfs(root1, seq1);
    dfs(root2, seq2);
    return seq1.toString() === seq2.toString();
};

相關文章