题目描述
给定两个非空二叉树 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。
思路
根据题意,如果s和t当前结点相同,则再此结点上进行DFS,因为是子树,所以要判断每一个结点是否匹配上;如果不相同,则看s得左子树和右子树得情况。
代码
/**
* 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) {
bool res = false;
if (s && t) {
if (s->val == t->val) res = dfs(s, t);
if (!res) res = isSubtree(s->left, t);
if (!res) res = isSubtree(s->right, t);
}
return res;
}
bool dfs(TreeNode* root1, TreeNode* root2) {
if (!root1 && !root2) return true;
if (!root1 || !root2) return false;
if (root1->val != root2->val) return false;
return dfs(root1->left, root2->left) && dfs(root1->right, root2->right);
}
};