golang 判断子树问题

最新推荐文章于 2025-04-28 22:40:53 发布

原创最新推荐文章于 2025-04-28 22:40:53 发布 · 243 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#golang #算法

golang 同时被 2 个专栏收录

9 篇文章

订阅专栏

算法

7 篇文章

订阅专栏

本文介绍了一种用于判断一棵树是否为另一棵树的子树的算法。通过定义TreeNode结构体并实现isSubtree和isEqual函数，可以有效地解决二叉树的子树判断问题。算法首先检查两棵树的根节点是否相等，然后递归地比较左子树和右子树。

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

func isSubtree(s *TreeNode, t *TreeNode) bool {
	if isEqual(s, t) {
		return true
	}

	if isEqual(s.Left, t) {
		return true
	}

	if isEqual(s.Right, t) {
		return true
	}

	return false
}

func isEqual(s *TreeNode, t *TreeNode) bool {
	if s == nil && t == nil {
		return true
	}

	if s != nil && t != nil && s.Val == t.Val {
		return isEqual(s.Left, t.Left) && isEqual(s.Right, t.Right)
	}

	return false
}