给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入: 1 1
/ \ / \
2 3 2 3
[1,2,3], [1,2,3]
输出: true
示例 2:
输入: 1 1
/ \
2 2
[1,2], [1,null,2]
输出: false
示例 3:
输入: 1 1
/ \ / \
2 1 1 2
[1,2,1], [1,1,2]
输出: false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/same-tree
两种方法解决:一是递归,二是非递归;
递归:思路:判断两棵树是否相同一是结构上相等二是结点元素值相等;可以用递归来解决此问题;
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if(p == NULL && q == NULL)
return true;
if((p == NULL && q != NULL) || (p != NULL && q== NULL))
return false;
if(p->val != q->val)
return false;
else
return isSameTree(p->left,q->left)&&isSameTree(p->right,q->right);
}
};
非递归实现:借用栈这个结构,用来保存每次需要进行回溯的结点;
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q)
{
stack<TreeNode*> pnode;
stack<TreeNode*> qnode;
if(p == NULL && q == NULL)
return true;
if((p == NULL && q != NULL) || (p != NULL && q== NULL))
return false;
if(p->val != q->val)
return false;
else
{
pnode.push(p);
qnode.push(q);
}
while(!pnode.empty() && !qnode.empty())
{
p = pnode.top();
q = qnode.top();
pnode.pop();
qnode.pop();
if(q == NULL&&p == NULL) continue;//如果两棵树当前的结点都为空,不做任何,跳到下一步,继续;
if((p==NULL&&q!=NULL)||(p!=NULL&&q==NULL)||(p->val != q->val)) return false;
pnode.push(p->left);
pnode.push(p->right);
qnode.push(q->left);
qnode.push(q->right);
}
if(pnode.empty() && qnode.empty())
return true;
return false;
}
};