leetcode 872. 叶子相似的树(Leaf-Similar Trees)

Posted 2021-02-15 zhanzq1

目录

• 题目描述：
• 解法：

题目描述：

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

举个例子，如上图所示，给定一颗叶值序列为 (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:
    void getList(TreeNode* root, vector<int>& lst){
        if(!root){
            return;
        }else if(!root->left && !root->right){
            lst.push_back(root->val);
        }else{
            getList(root->left, lst);
            getList(root->right, lst);
        }
    }
    
    bool equal(vector<int>& lst1, vector<int>& lst2){
        int sz = lst1.size();
        if(sz != lst2.size()){
            return false;
        }else{
            for(int i = 0; i < sz; i++){
                if(lst1[i] != lst2[i]){
                    return false;
                }
            }
            return true;
        }
    }
    
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> lst1, lst2;
        getList(root1, lst1);
        getList(root2, lst2);
        return equal(lst1, lst2);
    }
};