剑指Offer编程练习017——树的子结构

xiaoxiao2021-02-28  10

题目: 输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)/* 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* pRoot1, TreeNode* pRoot2)     {     } };

思路:先找到树A中与树B根节点rootB相同的节点rootA,然后比较rootA和rootB两棵树,完全相等就说明树B是树A的子结构。

class Solution { public: bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) { if(pRoot1==nullptr || pRoot2==nullptr) return false; if(pRoot1->val==pRoot2->val)//先找到根相等的 { if(IsSub(pRoot1,pRoot2))//然后判断这个根下面的子树是否相等 return true; } //先去左子树找,再去右子树找 return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right,pRoot2); } bool IsSub(TreeNode* p1,TreeNode* p2) { if(p2==nullptr) return true; if(p1==nullptr) return false; if(p1->val!=p2->val) return false; //左右子树必须严格匹配 return IsSub(p1->left,p2->left) && IsSub(p1->right,p2->right); } };
转载请注明原文地址: https://www.6miu.com/read-1650033.html

最新回复(0)