【LeetCode每日一题】938. 二叉搜索树的范围和

题目

请考虑一棵二叉树上所有的叶子，这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

举个例子，如上图所示，给定一棵叶值序列为 (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

题解

直接拼接字符串会卡下面测试用例：

[3,5,1,6,2,9,8,null,null,7,14]
[3,5,1,6,71,4,2,null,null,null,null,null,null,9,8]

原因是：7与14拼接得到714，71与4拼接也是714，实际是不相等的，解决这种问题直接再加个统一分隔符即可。

class Solution {
public:
    int index = 0;
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        string r1, r2;
        dfs1(root1, r1);
        dfs1(root2, r2);
        return r1 == r2;
    }
    void dfs1(TreeNode* root, string& res) {
        if (!root) return;
        if (!root->left && !root->right) {
            res += to_string(root->val) + ",";
            return;
        }
        dfs1(root->left, res);
        dfs1(root->right, res);
    }
   
};

上面效率太低了，给大家一个自己写的beat 100%代码，核心在于对第一次的结果存储下来后，后面直接比较并返回结果，而不是第二次遍历完毕才去比较。

class Solution {
public:
    int index = 0;
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> r1;
        dfs1(root1, r1);
        return dfs2(root2, r1) && index == r1.size();
    }
    void dfs1(TreeNode* root, vector<int>& res) {
        if (!root) return;
        if (!root->left && !root->right) {
            res.push_back(root->val);
            return;
        }
        dfs1(root->left, res);
        dfs1(root->right, res);
    }
    bool dfs2(TreeNode* root, vector<int>& res) {
        if (!root) return true;
        if (!root->left && !root->right) {
            if (index >= res.size()) return false;
            if (root->val != res[index]) return false;
            index++;
            return true;
        }
        return dfs2(root->left, res) && dfs2(root->right, res);
    }
   
};