LeetCode 872 叶子相似的树

题目

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

举个例子，如上图所示，给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。

如果有两颗二叉树的叶值序列是相同，那么我们就认为它们是 叶相似 的。

如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的，则返回 true；否则返回 false 。

• 提示：

  给定的两颗树可能会有 1 到 100 个结点。

我的代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> leaf;
    void dfs(TreeNode* root){
        if(root->left != NULL){
            dfs(root->left);
        }
        if(root->right != NULL){
            dfs(root->right);
        }
        if(root->left == NULL && root->right == NULL){
            leaf.push_back(root->val);
        }
        
    }
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> leaf1;
        vector<int> leaf2;
        dfs(root1);
        leaf1.swap(leaf);
        dfs(root2);
        leaf2.swap(leaf);
        int i =0;
        if(leaf1.size() == leaf2.size()){
            for(i=0;i<leaf1.size();i++){
                if(leaf1[i] != leaf2[i]){
                    return false;
                }
            }
            return true;
        }else{
            return false;
        }
    }
};

题解

两棵树分别深搜之后对储存的叶子数组