[E二叉树] lc872. 叶子相似的树(二叉树+dfs)

1. 题目来源

链接：872. 叶子相似的树

2. 题目解析

dfs 前序遍历下，将两棵树的叶子节点单独存下来即可。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

dfs

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root, vector<int>& q) {
        if (!root) return ;

        if (!root->left && !root->right) q.push_back(root->val);
        dfs(root->left, q);
        dfs(root->right, q);
        
    }
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> v1, v2;
        dfs(root1, v1);
        dfs(root2, v2);
        return v1 == v2;
    }
};