一、题目
网址:叶子相似的树。
难度:简单。
请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
二、程序
/**
* 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:
vector<int> leaf(TreeNode* root)
{
vector<int> leafs;
stack<TreeNode*> s;
if(root) s.push(root);
while(!s.empty())
{
TreeNode* p=s.top();
s.pop();
if(p->left==NULL&&p->right==NULL) leafs.push_back(p->val);
if(p->right!=NULL) s.push(p->right);
if(p->left!=NULL) s.push(p->left);
}
return leafs;
}
bool leafSimilar(TreeNode* root1, TreeNode* root2) {
return leaf(root1)==leaf(root2);
}
};
结果: