【LeetCode/力扣】872. 叶子相似的树

1 题目描述

题目链接：https://leetcode-cn.com/problems/leaf-similar-trees/

---

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

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

2 代码/C++/深度优先搜索dfs

/** 二叉树定义，由题目给出，每个节点包含值和左右两个子节点
 * 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:
	// dfs，深度优先搜索
	// 二叉树形参是二级指针，即指针的指针，两个指针指向同一地址
	// vector形参是引用，传递的是地址
    void dfs(TreeNode* node, vector<int>& vec){ 
        if (! node->left && !root->right){ // 当没有左子树和右子树，而该节点是叶节点
            vec.push_back(node->val); // push_back用于vector增加元素，pop_back()是对应删除
        }
        else{
            if (node->left){ // 如果左子树存在，则递归调用
                dfs(node->left, vec);
            }
            if (node->right){// 如果右子树存在，则递归调用
                dfs(node->right, vec);
            }
        }

    }
    
    bool leafSimilar(TreeNode* root1, TreeNode* root2) {
        vector<int> vec1, vec2; // 开辟两个vector容器依次保存两个二叉树的叶节点值
        dfs(root1, vec1); // 分别进行深度优先搜索
        dfs(root2, vec2);
        return vec1 == vec2; // 判断两个vector是否相同，相同则传回true，不同则传回false
    }
};