【算法】剑指 Offer 26. 树的子结构 和 面试题 04.10. 检查子树

输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构, 即 A中有出现和B相同的结构和节点值。

例如:
给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B:

   4 
  /
 1

代码:

func isSubStructure(A *TreeNode, B *TreeNode) bool {
	return (A != nil && B != nil) && (isSubStructure(A.Left, B) || isSubStructure(A.Right, B) || isSubStructureHelp(A, B))
}

func isSubStructureHelp(A *TreeNode, B *TreeNode) bool {
	if B == nil {
		return true
	}
	if A == nil {
		return false
	}
	return (A.Val == B.Val) && isSubStructureHelp(A.Right, B.Right) && isSubStructureHelp(A.Left, B.Left)
}

检查子树。你有两棵非常大的二叉树:T1,有几万个节点;T2,有几万个节点。设计一个算法,判断 T2 是否为 T1 的子树。

如果 T1 有这么一个节点 n,其子树与 T2 一模一样,则 T2 为 T1 的子树,也就是说,从节点 n 处把树砍断,得到的树与 T2 完全相同。

示例1:

 输入:t1 = [1, 2, 3], t2 = [2]
 输出:true
示例2:

 输入:t1 = [1, null, 2, 4], t2 = [3, 2]
 输出:false
提示:

树的节点数目范围为[0, 20000]。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-subtree-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

代码:

func checkSubTree(t1 *TreeNode, t2 *TreeNode) bool {
	if t1 == nil {
		return false
	}
	return isSubtreeHelp(t1, t2) || checkSubTree(t1.Right, t2) || checkSubTree(t1.Left, t2)
}

func isSubtreeHelp(s, t *TreeNode) bool {
	if s == nil && t == nil {
		return true
	}
	if s == nil || t == nil {
		return false
	}
	if s.Val == t.Val {
		return isSubtreeHelp(s.Right, t.Right) && isSubtreeHelp(s.Left, t.Left)
	}
	return false
}

*注意两道题目的区别,子树和子结构不一样

 

你可能感兴趣的:(算法,golang)