力扣-树-相同的数(LeetCode 100)
-
题目概述:给你两棵二叉树的根节点 p 和 q ,编写一个函数来检验这两棵树是否相同。(如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。)
-
解题思路:判断两树是否相同实际上是一个搜索遍历问题,一边遍历,一边判断。所以解决方法可分为深度优先搜索和广度优先搜索
-
深度优先搜索解题
- java代码
深度优先搜索这个方法最重要的就是递归的思想,但同时要考虑为空的情况
- java代码
class Solution{
public boolean isSameTree(TreeNode p, TreeNode q) {
//为空时要考虑两种情况,均为空还有一个空一个不空
if(p==null||q==null)return p==q;
if(p.val!=q.val) return false;
return isSameTree(p.left,q.left)&&isSameTree(p.right,q.right);
}
}
- python代码
class Solution:
def isSameTree(self, p: TreeNode, q: TreeNode) -> bool:
if(not p or not q):
return p==q
elif(p.val!=q.val):
return False
else:
return self.isSameTree(p.left,q.left) and self.isSameTree(p.right,q.right)
-
广度优先搜索解题
也就是前面做过的层次遍历,只是多加了比较两棵树的步骤,结尾记得比较队列是否为空,即可将其作为返回值。这个方法相比于深度搜索就有点繁琐了。- java代码
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if (p == null && q == null) {
return true;
} else if (p == null || q == null) {
return false;
}
Queue<TreeNode> queue1 = new LinkedList<TreeNode>();
Queue<TreeNode> queue2 = new LinkedList<TreeNode>();
queue1.offer(p);
queue2.offer(q);
while (!queue1.isEmpty() && !queue2.isEmpty()) {
TreeNode node1 = queue1.poll();
TreeNode node2 = queue2.poll();
if (node1.val != node2.val) {
return false;
}
TreeNode left1 = node1.left, right1 = node1.right, left2 = node2.left, right2 = node2.right;
if (left1 == null ^ left2 == null) {
return false;
}
if (right1 == null ^ right2 == null) {
return false;
}
if (left1 != null) {
queue1.offer(left1);
}
if (right1 != null) {
queue1.offer(right1);
}
if (left2 != null) {
queue2.offer(left2);
}
if (right2 != null) {
queue2.offer(right2);
}
}
return queue1.isEmpty() && queue2.isEmpty();
}
}