请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
解决方案
方法:深度优先搜索
思路和算法
首先,让我们找出给定的两个树的叶值序列。之后,我们可以比较它们,看看它们是否相等。
要找出树的叶值序列,我们可以使用深度优先搜索。如果结点是叶子,那么 dfs
函数会写入结点的值,然后递归地探索每个子结点。这可以保证按从左到右的顺序访问每片叶子,因为在右孩子结点之前完全探索了左孩子结点。
#include<iostream>
#include<vector>
using namespace std;
class solution
{
public:
typedef struct TNode
{
int data;
struct TNode* left;
struct TNode* right;
}TreeNode;
bool leafsimilar(TreeNode* root1, TreeNode* root2)
{
vector<int> leaves1;
vector<int> leaves2;
dfs(root1, leaves1);
dfs(root2, leaves2);
}
void dfs(TreeNode* node, vector<int>& leaves)
{
if(node == NULL)return;
if(node->left == NULL && node->right == NULL)
{
leaves.push_back(node->data);
}
dfs(node->left, leaves);
dfs(node->right, leaves);
}
};