题目描述
给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 1:
给定的树 s:
3 / \ 4 5 / \ 1 2
给定的树 t:
4 / \ 1 2
返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。
示例 2:
给定的树 s:
3 / \ 4 5 / \ 1 2 / 0
给定的树 t:
4 / \ 1 2
返回 false。
问题分析
先调用isSame函数,这个函数是判断两棵树是否完全相同。如果返回值为true,那么返回true。否则用s->left调用递归函数isSubtree,用s->right调用递归函数isSubtree,如果两个递归函数中至少有一个返回true,那么说明 t 树在以s->left为根节点的树中或在以s->right为根节点的树中,那么我们返回true,否则返回false。在上面调用两个递归函数之前,要先判断如果s节点为空,那么我们返回false。在isSame函数中,先判断如果两个节点都为空,那么返回true;否则其中有一个为空,另一个不为空返回false;接下来判断如果两个节点的值相同,并且用左孩子调用isSame函数返回true并且用右孩子调用isSame函数也返回true,那么说明两棵树相同,返回true,否则返回false。
代码实现
/**
* 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 isSubtree(TreeNode* s, TreeNode* t) {
if(isSame(s, t))
return true;
if(!s)
return false;
return isSubtree(s->left, t) || isSubtree(s->right, t);
}
bool isSame(TreeNode* s, TreeNode* t){
if(!s && !t)
return true;
if(!s || !t)
return false;
if(s->val == t->val && isSame(s->left, t->left) && isSame(s->right, t->right))
return true;
return false;
}
};