叶子相似的树
一、问题描述
请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举例:
如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。
提示:
给定的两颗树可能会有 1 到 100 个结点。
二、解题思路
-
利用先序遍历,递归得到每棵树的叶子结点
-
利用c++里的动态数组 创建两个数组 vector< int >test1,
vector< int >test2,将每棵树的叶子结点分别添加到test1,test2中 -
最后判断 test1==test2
三、核心代码
class Solution {
public:
bool leafSimilar(TreeNode* root1, TreeNode* root2)
{
if (root1 == NULL || root2 == NULL)
return false;
vector<int>test1;
vector<int>test2;
find(root1, test1);
find(root2, test2);
return test1 == test2;
}
void find(TreeNode* root, vector<int>& test3) {
if (root->left == NULL && root->right == NULL)
{
test3.push_back(root->val);
}
if (root->left) {
find(root->left, test3);
}
if (root->right) {
find(root->right, test3);
}
}
};