求树的子结构
好长时间不更新博客了,今天看到一道曾经做过的题(题目来自《剑指offer》),再练一下:
我的思路例如以下:
推断B是不是A的子结构。就是要推断B是不是和A中的一部分结构是不是全然同样。须要一个节点一个节点的推断。仅仅要有一个节点不同,就返回false。
这部分代码例如以下:
bool isEqual(TreeNode *head1,TreeNode *head2) { if(head2 == NULL) return true; if(head1 == NULL) return false; if(head1->data == head2->data) return isEqual(head1->left,head2->left) && isEqual(head1->right,head2->right); return false;<pre name="code" class="cpp">}
然后将A中的节点依次作为根节点同B的根节点进行比較:每一次比較都运行依次isEqual(),若isEqual()返回true,则总体返回true;否则将A的左子树和右子树同B比較。代码例如以下:
bool isSubTree(TreeNode *head1,TreeNode *head2) { if(head2 == NULL) return true; if(head1 == NULL) return false; if(isEqual(head1,head2)) return true; return isSubTree(head1->left,head2) || isSubTree(head1->right,head2); }