LeetCode算法题-- 相同的树
- 题目来源:https://leetcode-cn.com/problems/same-tree/
题目要求
给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入: 1 1
/ \ /
2 3 2 3
[1,2,3], [1,2,3]
输出: true
示例 2:
输入: 1 1
/ \ /
2 3 2 3
[1,2,3], [1,2,3]
输出: true
题目思路:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean isSameTree(TreeNode p, TreeNode q) {
if(p == null && q == null){
return true;
}
if(!checkTree(p,q)){
return false;
}
ArrayDeque<TreeNode> deqP = new ArrayDeque<TreeNode>();
ArrayDeque<TreeNode> deqQ = new ArrayDeque<TreeNode>();
deqP.addLast(p);
deqQ.addLast(q);
while(!deqP.isEmpty()){
p = deqP.removeFirst();
q = deqQ.removeFirst();
if(!checkTree(p,q)) return false;
if(p != null){//到这一步说明两个已经相等了,要确定是不是NULL
if(!checkTree(p.left, q.left)) return false;//同上
if(p.left != null){
deqP.addLast(p.left);
deqQ.addLast(q.left);
}
if(!checkTree(p.right,q.right)) return false;
if(p.right != null){
deqP.addLast(p.right);
deqQ.addLast(q.right);
}
}
}
return true;
}
public boolean checkTree(TreeNode p, TreeNode q){
if(p == null && q == null){ //原本以为这一步可以省略,但仔细一 想,如果这一步省略,下一个判断条件就会失效。
return true;
}
if(q == null || p == null){
return false;
}
if(p.val != q.val){
return false;
}
return true;
}
}
执行结果:
时间复杂度:o(n),n为节点数。