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 2 [1,2], [1,null,2] 输出: false
示例 3:
输入: 1 1 / \ / \ 2 1 1 2 [1,2,1], [1,1,2] 输出: false
解题关键:理解二叉树的基本操作。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int find(struct TreeNode *p1,struct TreeNode *p2,int *flag);
bool isSameTree(struct TreeNode* p, struct TreeNode* q) {
struct TreeNode* p1=p,*p2=q;
int flag = 1; //作为标记
find(p1,p2,&flag);
if(flag==1) return true;
else return false;
}
int find(struct TreeNode *p1,struct TreeNode *p2,int *flag){
if(*flag == 0) return 0;
if((p1!=NULL && p2==NULL )||(p1==NULL && p2!=NULL)){ //当一指针为空,另一指针非空则不同
*flag = 0;
return 0;
}
else if(p1 == NULL && p2 == NULL) return 0; //特殊情况返回
else{
if(p1->val != p2->val){ //若都非空则比较值
*flag = 0;
return 0;
}
find(p1->left,p2->left,flag); //进入二叉树下一层同一位置
find(p1->right,p2->right,flag);
}
return 0;
}
根据题解样例,优化代码:无需另写函数,只需将本函数不断递归
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
bool isSameTree(struct TreeNode* p, struct TreeNode* q) {
if(p==NULL && q ==NULL) return true;
if(p!=NULL && q!=NULL && p->val==q->val){
return isSameTree(p->left,q->left) && isSameTree(p->right,q->right);
}
return false;
}