版权声明:本文为博主原创文章,博客地址:https://blog.csdn.net/qq_41855420,未经博主允许不得转载。 https://blog.csdn.net/qq_41855420/article/details/91382132
请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (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:
bool leafSimilar(TreeNode* root1, TreeNode* root2) {
vector<int> firstTreeVec, secondTreeVec;
inorderTraver(root1, firstTreeVec);//中序遍历root1,将叶子节点的值按访问顺序放入firstTreeVec中
inorderTraver(root2, secondTreeVec);//中序遍历root2,将叶子节点的值按访问顺序放入secondTreeVec中
return firstTreeVec == secondTreeVec;//判断两个叶值序列是否相等
}
//中序遍历root,将叶子节点的值按访问顺序放入treeVec中
void inorderTraver(TreeNode* root, vector<int> &treeVec){
if (root == NULL){
return;
}
if (root->left == NULL && root->right == NULL){
//左右子树为空,说明是叶子节点
treeVec.push_back(root->val);
}
else{
inorderTraver(root->left, treeVec);
inorderTraver(root->right, treeVec);
}
}
};