LeetCode 872叶子相似的树-简单

最新推荐文章于 2021-05-10 16:46:18 发布

唐火

最新推荐文章于 2021-05-10 16:46:18 发布

阅读量97

点赞数 1

分类专栏： LeetCode

本文链接：https://blog.csdn.net/m0_51955470/article/details/115774090

版权

二叉树叶相似深度优先搜索节点值序列比较

关键词由CSDN通过智能技术生成

LeetCode 专栏收录该内容

31 篇文章 0 订阅

订阅专栏

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

在这里插入图片描述

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

代码如下:

/**
 * 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:
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int>l1;
        vector<int>l2;
        dfs(root1,l1);
        dfs(root2,l2);
        return l1==l2;
    }

    void dfs(TreeNode *root,vector<int>&v)
    {
        if (root==nullptr) return ;
        if (root->left==nullptr && root->right==nullptr) v.push_back(root->val);
        dfs(root->left,v);
        dfs(root->right,v);
    }
};