题目描述:
给定两个二叉树,编写一个函数来检验它们是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。
示例 1:
输入:
输出: true
示例 2:
输入:
输出: false
示例 3:
输入:
输出: false
方法1:递归判断
主要思路:
(1)比较简单,直接递归判断即可;
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
if(p==NULL&&q==NULL)//都为空,返回true
return true;
//结点处不满足要求的情形
if((!p&&q)||(p&&!q)||(p->val!=q->val))
return false;
//接着递归遍历
return isSameTree(p->left,q->left)&&isSameTree(p->right,q->right);
}
};
方法2:迭代判断
主要思路:
(2)使用两个栈分别存储两个树,使用先序遍历,比价各个结点的情形,若遍历的过程中出现了不符合要求的结果,则直接返回false,若遍历完整个树,则返回true;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
bool isSameTree(TreeNode* p, TreeNode* q) {
//判断根节点的情形
if(!p&&!q)
return true;
if((!p&&q)||(p&&!q)||(p->val!=q->val))
return false;
//分别存储两个树的栈
stack<TreeNode*> st_p;
stack<TreeNode*> st_q;
st_p.push(p);
st_q.push(q);
//存储过程是一样的,故只使用一个栈即可作为终止条件
while(!st_p.empty()){
//获得当前的要比较的两棵树的两个结点
p=st_p.top();
st_p.pop();
q=st_q.top();
st_q.pop();
//比较左结点
//比较满足要求的情形
if(p->left&&q->left&&p->left->val==q->left->val){
st_p.push(p->left);
st_q.push(q->left);
}
//不满足要求,要返回false的情形
else if((!p->left&&q->left)||(p->left&&!q->left)||(p->left&&q->left&&p->left->val!=q->left->val))
return false;
//比价右结点
if(p->right&&q->right&&p->right->val==q->right->val){
st_p.push(p->right);
st_q.push(q->right);
}
else if((!p->right&&q->right)||(p->right&&!q->right)||(p->right&&q->right&&p->right->val!=q->right->val))
return false;
}
//遍历结束,返回true
return true;
}
};