剑指Offer:树的子结构Java/Python

最新推荐文章于 2022-04-07 14:51:04 发布

原创最新推荐文章于 2022-04-07 14:51:04 发布 · 195 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否为另一棵二叉树A的子结构。通过递归方法，首先确认两树非空，然后检查B的根节点是否与A的根节点匹配，接着验证B的左右子树是否能与A的相应子树匹配。若直接匹配失败，则继续检查B是否为A的左子树或右子树的子结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

2.算法描述

首先A,B两棵树都不是空树才能比较，否则B不是A的子树。
然后，有四种情况：
a)B的根与A的根对应上了，就去比较B的左右子树与A的左右子树是否可以对应上。
b)如果，B不直接是A的子树，看B是不是A的左子树的子树。
c)如果，B也不是A左子树的子树，看B是A的右子树的子树。
d)B确实不是A的子树。

3.代码描述

3.1.Java代码

/*
public class TreeNode{	
	int val;
	TreeNode left = null;
	TreeNode right = nul;
	public TreeNode(int val){
		this.val = val;
	}
}
*/
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        boolean ans = false;
        if(root1!=null && root2!=null){//两棵树都不是空树 就进行比较
            if (root1.val==root2.val)//如果两棵树的根对应了，直接看root2是不是root1的子树
                ans = isSubtree(root1, root2);
            if(ans==false)//看root2是不是root1左子树的子树
                ans = HasSubtree(root1.left, root2);
            if(ans==false)//看root2是不是root1右子树的子树
                ans = HasSubtree(root1.right, root2);
        }
        return ans;
    }
    private boolean isSubtree(TreeNode root1, TreeNode root2){
        if(root2==null)//先看root2是不是比较完了，比较完了 说明root2与root1是对应的上的  是root1的子树
            return true;
        if(root1==null)//如果root还没有比较完， 然是root1却比较完了，说明root2不是root1的子树
            return false;
        if(root1.val!=root2.val)//如果对应结点的值不等  不会是子树
            return false;
        return isSubtree(root1.left, root2.left) && isSubtree(root1.right, root2.right);
    }
}

3.2.Python代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        ans = False
        if pRoot1!=None and pRoot2!=None:
            if pRoot1.val == pRoot2.val:
                ans = self.isSubtree(pRoot1, pRoot2)
            if ans==False:
                ans = self.HasSubtree(pRoot1.left, pRoot2)
            if ans==False:
                ans = self.HasSubtree(pRoot1.right, pRoot2)
        return ans
    
    def isSubtree(self, r1, r2):
        if r2==None:
            return True
        if r1==None:
            return False
        if r1.val!=r2.val:
            return False
        else:
            return self.isSubtree(r1.left, r2.left) and self.isSubtree(r1.right, r2.right)