剑指 Offer 26：树的子结构

题目

题目链接
输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2

给定的树 B：

   4 
  /
 1

返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false

示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

解题

和leetcode-572：另一棵树的子树是一样的
思路都是,遍历每个节点，如果A的节点的值，和B的头节点值相等，就判断两棵树是否相等。

方法一：递归(先序遍历)

class Solution {
    
    
private:
    bool isSame(TreeNode* A,TreeNode* B){
    
    
        if(!B) return true;//只要B空了，就说明B都遍历完了，就可以返回true
        if(!(A&&B)) return false;//如果A为空，B不为空，说明不相同
        if(A->val!=B->val) return false;
        return isSame(A->left,B->left)&&isSame(A->right,B->right);
    }
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
    
    
        if(!(A&&B)) return false;//A和B任意一个为空，都代表B节点不可能是A的子节点
        if(A->val==B->val&&isSame(A,B)) return true;
        return isSubStructure(A->left,B)||isSubStructure(A->right,B);
    }
};

方法二：递归和迭代

class Solution {
    
    
private:
    bool isSame(TreeNode* A,TreeNode* B){
    
    
        if(!B) return true;
        if(!(A&&B)) return false;
        if(A->val!=B->val) return false;
        return isSame(A->left,B->left)&&isSame(A->right,B->right);
    }
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
    
    
        if(!(A&&B)) return false;
        queue<TreeNode*> queue;
        queue.push(A);
        while(!queue.empty()){
    
    
            TreeNode* cur=queue.front();
            queue.pop();
            if(cur->val==B->val&&isSame(cur,B)) return true;
            if(cur->left) queue.push(cur->left);
            if(cur->right) queue.push(cur->right);
        }
        return false;
    }
};