3-树-相同的树

这是树的第3篇算法，力扣链接。

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1：

输入：p = [1,2,3], q = [1,2,3]
输出：true

示例 2：

输入：p = [1,2], q = [1,null,2]
输出：false

这道题最简单的做法就是递归，也就是传说中的深度优先遍历。

func isSameTree(p *TreeNode, q *TreeNode) bool {
	if p == nil && q == nil {
		return true
	}
	if p == nil || q == nil {
		return false
	}
	if p.Val != q.Val {
		return false
	}
	return isSameTree(p.Left, q.Left) && isSameTree(p.Right, q.Right)
}

 然后可以写成迭代方法：

func isSameTree(p *TreeNode, q *TreeNode) bool {
	stack1, stack2 := []*TreeNode{p}, []*TreeNode{q}
	for len(stack1) > 0 && len(stack2) > 0 {
		rootp := stack1[len(stack1)-1]
		rootq := stack2[len(stack2)-1]
		stack1 = stack1[:len(stack1)-1]
		stack2 = stack2[:len(stack2)-1]
		if rootp == nil && rootq == nil {
			continue
		}
		if rootp == nil || rootq == nil || rootp.Val != rootq.Val {
			return false
		}
		stack1 = append(stack1, rootp.Right)
		stack1 = append(stack1, rootp.Left)
		stack2 = append(stack2, rootq.Right)
		stack2 = append(stack2, rootq.Left)
	}
	return len(stack1) == 0 && len(stack2) == 0
}

现在的做法是模拟stack的做法，后入先出的方法遍历这棵树，当然也可以模拟queue，先入先出，这就成了广度优先遍历的逻辑。

func isSameTree(p *TreeNode, q *TreeNode) bool {
	queue1, queue2 := []*TreeNode{p}, []*TreeNode{q}
	for len(queue1) > 0 && len(queue2) > 0 {
		rootp := queue1[0]
		rootq := queue2[0]
		queue1 = queue1[1:]
		queue2 = queue2[1:]
		if rootp == nil && rootq == nil {
			continue
		}
		if rootp == nil || rootq == nil || rootp.Val != rootq.Val {
			return false
		}
		queue1 = append(queue1, rootp.Right)
		queue1 = append(queue1, rootp.Left)
		queue2 = append(queue2, rootq.Right)
		queue2 = append(queue2, rootq.Left)
	}
	return len(queue1) == 0 && len(queue2) == 0
}