【简单】872. 叶子相似的树

最新推荐文章于 2021-12-20 13:59:59 发布

菜菜2022

最新推荐文章于 2021-12-20 13:59:59 发布

阅读量161

点赞数

分类专栏：刷题 # leetcode

本文链接：https://blog.csdn.net/kz_java/article/details/106663941

版权

leetcode 同时被 2 个专栏收录

714 篇文章 5 订阅

订阅专栏

刷题

643 篇文章 5 订阅

订阅专栏

【题目】
请考虑一颗二叉树上所有的叶子，这些叶子的值按从左到右的顺序排列形成一个叶值序列。
在这里插入图片描述举个例子，如上图所示，给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。如果有两颗二叉树的叶值序列是相同，那么我们就认为它们是叶相似的。如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的，则返回 true；否则返回 false。
来源：leetcode
链接：https://leetcode-cn.com/problems/leaf-similar-trees/
【提示】
给定的两颗树可能会有 1 到 200 个结点
给定的两颗树上的值介于 0 到 200 之间
【代码】
执行用时 :0 ms, 在所有 C++ 提交中击败了100.00% 的用户
内存消耗 :11.3 MB, 在所有 C++ 提交中击败了100.00%的用户

/**
 * 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 {
private:
    stack<TreeNode*> s;
public:
    void deal(TreeNode*node1,vector<int> &v){
        s.push(node1);
        while(!s.empty()){
            node1=s.top();
            s.pop();
            if(node1->left==NULL&&node1->right==NULL)
                v.push_back(node1->val);
            if(node1->left)
                s.push(node1->left);
            if(node1->right)
                s.push(node1->right);
        }
    }
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> v,v2;
        deal(root1,v);
        deal(root2,v2);
        for(int i=0;i<v.size();i++)
            if(v[i]!=v2[i]||v.size()!=v2.size())
                return false;
        return true;
    }
};

【繁琐】

/**
 * 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:
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> v;
        stack<TreeNode*> s;
        TreeNode* node1=root1;
        s.push(node1);
        while(!s.empty()){
            node1=s.top();
            s.pop();
            if(node1->left==NULL&&node1->right==NULL)
                v.push_back(node1->val);
            if(node1->left)
                s.push(node1->left);
            if(node1->right)
                s.push(node1->right);
        }
        node1=root2;
        int cnt=0;
        s.push(node1);
        while(!s.empty()){
            node1=s.top();
            s.pop();
            if(node1->left==NULL&&node1->right==NULL)
                if(v[cnt++]!=node1->val)
                    return false;
            if(node1->left)
                s.push(node1->left);
            if(node1->right)
                s.push(node1->right);
        }
        return true;
    }
};