力扣刷题（剑指 Offer 26. 树的子结构）回溯

剑指 Offer 26. 树的子结构 - 力扣（LeetCode）

输入两棵二叉树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

思路：

整体的大思路就是回溯，判断A和B在该节点是否相等，相等继续去判断左右子树是否均与B的左右子树对应，不相等就再分别判断A->left、B->left与A->right、B->righ是否满足要求。主要就是要加上一个是否已经开始比较的状态，如果开始了，那本次比较的时候值不相等，说明开始比较的节点就不成立，直接返回false。

还有就是要注意空子树不为任何树的子结构

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
//整体上就是一个回溯的思想，我先判断这个节点下A和B的val是否相等，如果相等，就去查看其左右节点
//如果不相等就要看目前的状态已经处于比较中还是没有开始比较；
//已经开始比较了，直接返回false
//还没有开始比较，那就去递归判断A\B的左右子树是否分别满足条件
class Solution {
public:
    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if(!A||!B)return false;
        return is_sub(A,B,false);
    }
    
    bool is_sub(TreeNode* A, TreeNode* B,bool is_start){
        if(!A||!B)return false;
        bool bool_1=false;
        if(A->val==B->val){
            bool_1=(B->left?is_sub(A->left,B->left,true):true)&&(B->right?is_sub(A->right,B->right,true):true);
        }
        else if(is_start)return false;
        if(!bool_1)bool_1=is_sub(A->left,B,false)||is_sub(A->right,B,false);
        return bool_1;
    }
};