题目链接
题目描述
请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一棵叶值序列为 (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,2,3], root2 = [1,3,2] 输出:false
提示:
- 给定的两棵树结点数在
[1, 200]
范围内 - 给定的两棵树上的值在
[0, 200]
范围内
思路分析
知识点:递归、深度优先搜索、二叉树
解析:
我们可以通过深度优先搜索(dfs)算法,找到树的所有叶子结点并放到数组里面,之后将这个数组进行比较,返回false的条件有两个,取其一即可:
1.两数组长度不等
2.值不相等
递归细节:如果左右孩子都为空,说明递归到叶子结点了,就将这个结点的值放入数组,如果不是,如果左孩子存在,就递归到左孩子,如果右孩子存在,就递归到右孩子。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
void dfs(struct TreeNode* root,int* seq,int* seqsize)//深度优先搜索算法
{
if(root->left==NULL&&root->right==NULL)
{
seq[(*seqsize)++]=root->val;
}
else
{
if(root->left)
{
dfs(root->left,seq,seqsize);
}
if(root->right)
{
dfs(root->right,seq,seqsize);
}
}
}
bool leafSimilar(struct TreeNode* root1, struct TreeNode* root2) {
int seq1[200]={0};
int seqsize1=0;
if(root1)
{
dfs(root1,seq1,&seqsize1);
}
int seq2[200]={0};
int seqsize2=0;
if(root2)
{
dfs(root2,seq2,&seqsize2);
}
if(seqsize1!=seqsize2)
{
return false;
}
for(int i=0;i<seqsize1;i++)
{
if(seq1[i]!=seq2[i])
{
return false;
}
}
return true;
}