1.题目
输入两棵二叉树A,B,判断B是不是A的子结构。(我们约定空树不是任意一个树的子结构)
假如给定A为{8,8,7,9,2,#,#,#,#,4,7},B为{8,9,2},2个树的结构如下,可以看出B是A的子结构
2.解法
2.1 树的结构
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
2.2 解法
两棵二叉树A,B,A如果包含B,说明内部有一个子结构与B完全相同。
那么先实现相同的判定,如果根节点开始不相同就考虑左孩子和右孩子是否与B相同。 这里存在递归调用。
一层递归是对HasSubtree
的方法的调用是递归调用,一是sameTree
的递归调用。
注意&& 和 || 的区别,在sameTree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2)
语句里,需要考虑有一个实现即可。
而sameTree(node1.left, node2.left) && sameTree(node1.right, node2.right)
则需要左右孩子完全相同。
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
if (root1 == null || root2 == null) return false;
return sameTree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
}
public boolean sameTree(TreeNode node1, TreeNode node2) {
if (node1 == null && node2 == null) return true;
if (node1 == null) return false;
if (node2 == null) return true;
if (node1.val == node2.val) {
return sameTree(node1.left, node2.left) && sameTree(node1.right, node2.right);
}
return false;
}
总结
这个递归有一点点绕,但是看了后就会发现挺有意思的。
算法系列在github上有一个开源项目,主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客,将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。