872. 葉子相似的樹 樹 遍歷葉子
請考慮一顆二叉樹上所有的葉子,這些葉子的值按從左到右的順序排列形成一個 葉值序列 。
舉個例子,如上圖所示,給定一顆葉值序列為 (6, 7, 4, 9, 8)
的樹。
如果有兩顆二叉樹的葉值序列是相同,那麼我們就認為它們是 葉相似 的。
如果給定的兩個頭結點分別為 root1
和 root2
的樹是葉相似的,則返回 true
;否則返回 false
。
/**
* 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;
findLeave(root1,r1);
findLeave(root2,r2);
int lr1=r1.size();
int lr2=r2.size();
if(lr1!=lr2)return false;
for(int i=0;i<lr1;i++)
if(r1[i]!=r2[i])return false;
return true;
}
void findLeave(TreeNode* t,vector<int> &r)
{
if(t->left==NULL&&t->right==NULL)
{
r.push_back(t->val);
return;
}
if(t->left!=NULL)
findLeave(t->left,r);
if(t->right!=NULL)
findLeave(t->right,r);
}
};