《树》剑指 Offer 26. 树的子结构《leetcode》

剑指 Offer 26. 树的子结构

难度中等151

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

限制：

0 <= 节点个数 <= 10000

思路解析：

  思路1.先找到与子树头结点一致的结点，存储到数组中，依次判断其是否与子树一致，如一致，返回true，如都不一致，返回false。

  思路2.逐次比较每个节点即可。

思路1AC代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<TreeNode>node=new ArrayList();

    public void findSubRoot(TreeNode A,int bVal)//dfs;
    {
        if(A==null)
            return;
        if(A.val==bVal)
            node.add(A);
        findSubRoot(A.left,bVal);
        findSubRoot(A.right,bVal);
    }

    public boolean isSameTree(TreeNode A,TreeNode B)
    {
        if(B==null)//B是子树 B只要是null，无论A是否是null都匹配
            return true;
        if(A==null)//A=null 只有B是null时，才true 但是被前面B的判断过滤掉，所以只返回false即可。
            return false;
        if(A.val!=B.val)
            return false;
        return isSameTree(A.left,B.left)&&isSameTree(A.right,B.right);
    }

    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(B==null)
            return false;
        
        findSubRoot(A,B.val);//node被更新

        for(int i=0;i<node.size();i++)
            if(isSameTree(node.get(i),B))
                return true;
        return false;
    }
}

思路2AC代码：

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

再提一点:

  树的遍历愈发熟练了，下次搞搞非递归的。