题解 | #树的子结构#
树的子结构
http://www.nowcoder.com/practice/6e196c44c7004d15b1610b9afca8bd88
树的子结构
思路
1、原始函数的功能是:判断root1的树中是否存在root2,因此当遇到节点值不相等的时候,判断root1的左子树是否包含root2 或者 root1的右子树是否包含root2的树(递归)
2、辅助函数的功能:判断以当前节点为根的树是否相等,若节点相等,则继续比较对应的左右节点是否相等(递归)
代码
/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } };*/ class Solution { public: bool HasSubtree(TreeNode* root1, TreeNode* root2){ if(root2==nullptr){ return false; } if(root1==nullptr){ return false; } if(isPart(root1, root2)){ return true; } // 不是一部分 return HasSubtree(root1->left, root2) || HasSubtree(root1->right, root2); } // 以root1起,后面是否是相等的 bool isPart(TreeNode* pRoot1, TreeNode* pRoot2) { if(pRoot2==nullptr){ return true; } if(pRoot1==nullptr){ return false; } if(pRoot1->val != pRoot2->val){ return false; } return isPart(pRoot1->left, pRoot2->left) && isPart(pRoot1->right, pRoot2->right); } };