给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 1:
给定的树 s:
3
/ \
4 5
/
1 2
给定的树 t:
4
/
1 2
返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。
示例 2:
给定的树 s:
3
/
4 5
/
1 2
/
0
给定的树 t:
4
/
1 2
返回 false。
解法
public boolean isSubtree(TreeNode s, TreeNode t) {
if(s == null) return false;
if(isSame(s, t))
return true;
else
return isSubtree(s.left, t) || isSubtree(s.right, t);
}
private boolean isSame(TreeNode p, TreeNode q){
if(q == null && p == null) return true;
if(q == null || p == null) return false;
if(q.val != p.val) return false;
else return isSame(q.left, p.left) && isSame(q.right, p.right);
}
思路分析:
题目所指的s中是否包含t,就是在问s中是否有某一棵子树与t相同(相等就是树结构一致,且每个结点值都相等。)
如何判断两棵树是否相同,辅助函数boolean isSame(TreeNode p, TreeNode q),可以参考仓库里题解100. 相同的树。
根节点值相同,且左右子树相同才能判断一颗树相同。这就是递归的关系。
递归结束的条件,当两颗树均是null时他们相等,当两颗树其中之一为null另外一颗非null时他们不相等。
其它情况下,先判断当前节点的值是否相等,如果相等再判断当前节点的左右子树是否均相同
再回到树的包含问题上
如果结点s与t为根的树相同,显然s包含t。 if(isSame(s, t)) return true;。
否则判断s的左子树或者右子树是否包含t,如果有其中之一包含t,也说明s包含t,所以返回isSubtree(s.left, t) || isSubtree(s.right, t);
当递归中s == null,显然这棵空树不包含t,且没有子树可以进行递归,直接返回false。