树的子结构

题目链接

思路

• 比较A,B两个节点,如果根不同就拿A.Left和A.Right与B对比
• 如果A,B两个根节点相同,就拿A.Left和B.Left;A.Right和B.Rught对比;直至将B对比完毕返回true
• 如果将A对比完毕都没有与B相同的结构返回false

java

    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (A == null || B== null){
            //约定空树不是任意一个树的子结构
            return false;
        }
        // A的根节点或左右节点任意一个节点与B相等都返回true
        return isSub(A,B) || isSubStructure(A.left,B) || isSubStructure(A.right,B);
    }

    private boolean isSub(TreeNode a, TreeNode b) {
        if (b == null){
            // b对比完成
            return true;
        }
        if (a == null){
            // 对比未完成,a已经未空
            return false;
        }
        if (a.val != b.val){
            // 两个节点的值不等;对比结束
            return false;
        }
        // 分别对比左右节点
        return isSub(a.left,b.left) && isSub(a.right,b.right);
    }

go

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func isSubStructure(A *TreeNode, B *TreeNode) bool {
	if A == nil || B == nil {
		return false
	}
	// A的根节点或左右节点任意一个节点与B相等都返回true
	return isSub(A, B) || isSubStructure(A.Left, B) || isSubStructure(A.Right, B)
}

func isSub(a, b *TreeNode) bool {
	if b == nil {
		// b对比完成
		return true
	}
	if a == nil {
		// 对比未完成,a已经未空
		return false
	}
	if a.Val != b.Val {
		// 两个节点的值不等;对比结束
		return false
	}
	// 分别对比左右节点
	return isSub(a.Left, b.Left) && isSub(a.Right, b.Right)
}