题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
牛客上的一道关于树的题,觉得自己在树这一块有点弱鸡,写下来加深一下印象
分析:判断B树是不是A树中的一部分,可以理解为。(他们说山东人特别喜欢倒装,看来是的)
思路:首先判断B树的头结点与A树的头结点是否为空,空的话就直接false了,如果不为空,判断是否相等,
如果相等的话,在判断左右节点是否相等,直到B的所有节点都遍历完了,才说明B树是A的子结构,如果没遍历完(就是说B树的左右节点还有值),并且发现此时A节点的值跟此时B节点的值还正好相等,我们继续吧A节点的左右节点与B节点的左右节点在放进去比较,直到B树为空或者期间出现了不相等的情况。
如果不相等呢,我们就把A树的节点移动到左节点,还不行就移动到右节点。
代码:
public static boolean HasSubtree(TreeNode root1,TreeNode root2){
boolean result=false;
if (root1!=null&&root2!=null) {
if (root1.val==root2.val) {
result=doesTree1HaveTree2(root1, root2);
}
if (!result) {
result=HasSubtree(root1.left, root2);
}
if (!result) {
result=HasSubtree(root1.right, root2);
}
}
return result;
}
public static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
//Tree2遍历完了
if (node2==null) {
return true;
}
//Tree1遍历完了
if (node1==null) {
return false;
}
if (node1.val!=node2.val) {
return false;
}
return doesTree1HaveTree2(node1.left, node2.left)&&doesTree1HaveTree2(node1.right, node2.right);
}