leetcode872:葉子相似的樹

---

---

一、題目

網址：葉子相似的樹。

難度：簡單。

請考慮一顆二叉樹上所有的葉子，這些葉子的值按從左到右的順序排列形成一個 葉值序列 。

舉個例子，如上圖所示，給定一顆葉值序列爲 (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);
    }
};

結果：