Leetcode--另一個樹的子樹

給定兩個非空二叉樹 s 和 t，檢驗 s 中是否包含和 t 具有相同結構和節點值的子樹。s 的一個子樹包括 s 的一個節點和這個節點的所有子孫。s 也可以看做它自身的一棵子樹。

示例 1:
給定的樹 s:

     3
    / \
   4   5
  / \
 1   2
給定的樹 t：

   4 
  / \
 1   2
返回 true，因爲 t 與 s 的一個子樹擁有相同的結構和節點值。

示例 2:
給定的樹 s：

     3
    / \
   4   5
  / \
 1   2
    /
   0
給定的樹 t：

   4
  / \
 1   2
返回 false。

思路

參考Leetcode題解；

先找到值相等的兩個節點，然後判斷以這兩個節點的子樹是否相等，

如果子樹相等則返回ture，否則繼續尋找相等的節點

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    TreeNode* originT;
    // 判斷兩個子樹是否相同
    bool isSame(TreeNode* s, TreeNode* t){
        // 兩個節點都爲空，則返回true
        if(!t&& !s) return true;
        // 有個爲空，說明兩個節點不等
        if(!s|| !t) return false;
        // 當前節點相等且左右子樹都相等，則返回true
        if(s->val== t->val)
            return isSame(s->left, t->left)&& isSame(s->right, t->right); 
        else
            return false;
    }
    bool isSubtree(TreeNode* s, TreeNode* t) {
        if(!s) return false;
        // 當前節點相等且兩個子樹相等
        else if(s->val== t->val&& isSame(s, t))
            return true;
        else
            return isSubtree(s->left, t)|| isSubtree(s->right, t);
    }
};