572. 另一个树的子树
给定两个非空二叉树 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。
解法一
/**
* 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 comp(TreeNode* s, TreeNode* t) {
if(!s && !t) return true;
if(!s && t || s && !t) return false;
if(s->val != t->val) return false;
return comp(s->left, t->left) && comp(s->right, t->right);
}
bool isSubtree(TreeNode* s, TreeNode* t) {
if(!s && !t) return true;
if(!s && t || s && !t) return false;
if(comp(s, t)) return true;
return isSubtree(s->left, t) || isSubtree(s->right, t);
}
};
思路:
双重递归。主函数isSubtree递归地遍历s的每个结点,在遍历过程中,比较以当前结点为根的树与树t比较;函数comp则比较两个树是否完全相等。
2019/06/02 23:07