剑指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

思路

采用递归思路，将问题分解成两个

• 遍历A树的所有节点
• 判断以每个结点为根节点是否和B树有相同结构

代码

class Solution {
   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);
		  	
	    }
	  
	  
	  public boolean dfs(TreeNode A,TreeNode B) {
		  if(B==null) { return true; }
		  if(A==null||A.val!=B.val) {return false;}
		  return dfs(A.left,B.left) && dfs(A.right , B.right);
	  }
	  
}