今日题目(剑指Offer系列)
剑指 Offer 26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构) B是A的子结构, 即 A中有出现和B相同的结构和节点值。 例如: 给定的树 A:
示例:
示例 1: 输入:A = [1,2,3], B = [3,1] 输出:false 示例 2: 输入:A = [3,4,5,1,2], B = [4,1] 输出:true
解题思路:
>给问题是说判断B是不是A的子结构 >其实就是遍历A看能不能找出B的结构 >首先遍历A,将A的根节点与B进行判断 >如果不行将A的左右子树分别与B进行判断,递归下去 >判断函数就是判断两个树对应节点的值是是否对应相等
Python解法:
class Solution: def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool: def judge(A,B): if B==None: return True if A==None: return False if A.val!=B.val: return False return judge(A.left,B.left) and judge(A.right,B.right) if A==None or B==None: return False return judge(A,B) or self.isSubStructure(A.left,B) or self.isSubStructure(A.right,B)
Java解法:
class Solution { public boolean isSubStructure(TreeNode A, TreeNode B) { if(A==null||B==null){ return false; } return judge(A,B) || isSubStructure(A.left,B) || isSubStructure(A.right,B); } public boolean judge(TreeNode A,TreeNode B){ if(B==null){ return true; } if(A==null){ return false; } if(A.val!=B.val){ return false; } return judge(A.left,B.left)&&judge(A.right,B.right); } }