请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
提示:
- 给定的两颗树可能会有
1
到100
个结点。
AC:
/**
* 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> r1, r2; //记录叶子序列
f(root1,r1);
f(root2,r2);
if(r1.size() == r2.size())
{
for(int i = 0; i < r1.size(); i++)
{
if(r1[i] != r2[i]) return false;
else continue;
}
return true;
}else return false;
}
void f(TreeNode* root, vector<int>& r)
{
if(root->left == NULL && root->right == NULL)
{
r.push_back(root->val);
}
if(root->left) f(root->left,r);
if(root->right) f(root->right,r);
}
};