C++Leetcode872:叶子相似的树

题目
请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
在这里插入图片描述
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。

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

思路
1、递归。分别递归2个二叉树,将叶子节点分别存到2个动态数组中,最后判断这两个数组是否相等。

实现方法
一、递归

/**
 * 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> v1,v2;
        leafList(root1,v1);
        leafList(root2,v2);
        return v1 == v2 ? true:false;
    }
    vector<int> leafList(TreeNode* root,vector<int> &v){
        if(!root) return v;
        if(!root->left && !root->right){
            v.push_back(root->val);
        }
        if(root->left) leafList(root->left,v);
        if(root->right) leafList(root->right,v);
        return v;
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43434305/article/details/87876584