LeetCode——剑指 Offer 26. 树的子结构

剑指 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）
链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解思路

1>源代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(B==null || A==null){
            return false;
        }
        return helper(A,B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    private boolean helper(TreeNode A, TreeNode B){
        if(B==null) return true;
        if(A==null || A.val!=B.val) return false;
        return helper(A.left, B.left) && helper(A.right, B.right);
    }
}

2>算法介绍

本题乍一看感觉好像很简单，只需要对两个树进行先序遍历，并比对先序遍历结果即可。如果发现不匹配则检测A的左子树和A的右子树是否符合。很明显，这是一道递归的题。

但是本题的递归函数需要单独完成，并不能用题目给的isSubStructure()，原因是本题的边界判定条件比较特殊。题目规定空树不作为任何树的子树，所以在主函数里如果遇到两个输入的其中之一为空，则应该立刻返回false。但对于递归来说，如果A树为空B不为空，说明B不是A的子树，返回false; 但如果B树为空，则说明已经比对完成，这时应该返回true！

想明白了这一点之后，按照先序遍历的框架完成递归即可。