比较两个二叉树是否相同。
我们很容易想到这就是一个遍历树的变形的题目,于是我写了下面的代码:
ps:这个代码是错的!!!!
bool isSameTree(TreeNode* p, TreeNode* q) {
if(p != NULL && q!=NULL){
if(p->val == q->val){
return isSameTree(p->left,q->left);
return isSameTree(p->right,q->right);
}else{
return false;
}
}else if(p == NULL && q != NULL){
return false;
}else if(p != NULL && q == NULL){
return false;
}
return true;
}
很显然我用了前序遍历,但我在如何返回true这里完全找不到合适的方法,我这写只需要某一个子支是相同的,那我就会提前返回true导致wrong answer。这里我想要写成必须考虑所有情况都相同才返回true,但我一直写不清(想不清)这个逻辑,无奈偷看了网上大神的代码,看完之后叹为惊人…
AC代码:
bool isSameTree(TreeNode* p, TreeNode* q) {
if(p ==NULL && q == NULL){
return true;
}else if(p == NULL || q== NULL){
return false;
}else{
return p->val == q->val && isSameTree(p->left,q->left) && isSameTree(p->right,q->right);
}
}
这样除非是根就都是空的,其他的时候某一个支条到了尽头只是完成了返回true的1/3的条件。
所以说编程始终是智商的游戏,而我的智商看来是相当堪忧又难以提高的,这让我很惆怅啊…