开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第8天,点击查看活动详情
一、题目描述:
给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:p = [1,2,3], q = [1,2,3]
输出:true
示例 2:
输入:p = [1,2], q = [1,null,2]
输出:false
示例 3:
输入:p = [1,2,1], q = [1,1,2]
输出:false
提示:
- 两棵树上的节点数目都在范围 [0, 100] 内
- -10^4 <= Node.val <= 10^4
二、思路分析:
两颗树要相等,只有两种情况:
1、都为空树。
2、他们的根节点值相等,并且其左右子树也相等。其他情况则必然不等。
三、AC 代码:
func isSameTree(p *TreeNode, q *TreeNode) bool {
switch {
case p == q && p == nil:
return true
case p != nil && q != nil:
return p.Val == q.Val && isSameTree(p.Left, q.Left) && isSameTree(p.Right, q.Right)
default:
return false
}
}
四、总结:
遇到树,一般都是使用递归来解决:
- 无非是递归左,递归右,结束条件是只要出现value不相等,下面就不需要递归了,直接返回
- 如果其中一个树相同深度走完了,而另一个没有,那么可以返回false了
- 两个都走完了,返回true,说明是符合的
另外
二叉树对比无非就是分别判断 val, isSameTree(left), isSameTree(right)直到遍历到left, right都是 Null 为止,因此如果两个都是null相等,一个是null另一个不是不相等,值不一样不相等,值一样相等,分别判断即可
范文参考: