872. 叶子相似的树(C++)

题目

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

image-20200902200037659

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

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

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

提示:

  • 给定的两颗树可能会有 1 到 200 个结点。
  • 给定的两颗树上的值介于 0 到 200 之间。

分析与题解

定义两个数组用于获取叶结点值。代码中函数DFS()实现这个功能。、

判断当前结点是不是叶结点,是的话把值压进数组,不是的话寻找递归该结点的左右子树。

因为最终只需要返回一个vector中的叶值序列,所以每次递归中push_back的值都需要保留下来,因此采用引用传递传递参数。

/**
 * 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:
    //形参以引用形式进行迭代
    void DFS(TreeNode* root, vector<int> &res){
        if(root==NULL)
            return;
        if(root->left==NULL && root->right==NULL) 
            res.push_back(root->val);
        DFS(root->left,res);
        DFS(root->right,res);
    }
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        if(root1==NULL && root2==NULL)
            return true;
        else if(root1==NULL || root2==NULL)
            return false;
        vector<int> res1,res2;
        DFS(root1,res1);
        DFS(root2,res2);
        //因为是正序排列的,所以如果叶值序列相同,那么其从左至右的顺序应该也一致
        return res1==res2;

    }
};
posted @ 2020-09-02 20:04  脱线森林`  阅读(119)  评论(0编辑  收藏  举报