1. 程式人生 > >872. 葉子相似的樹 樹 遍歷葉子

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);
    }
};