树的子结构

树的子结构

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

思路

判断A树中当前结点是否与B树的根结点相同，如果不相同，接着遍历A树；如果相同，则A树从当前结点与B树从根结点开始同步遍历，直至B树遍历完且遍历过程中A、B两树对应结点都相同，则B树是A树的子结构；否则返回，重新在A树中寻找与B树根结点相同的结点，接着遍历。

代码（c++）

/* struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; TreeNode(int x) : val(x), left(NULL), right(NULL) { } };*/ class Solution { public: bool judge(TreeNode* pRoot1, TreeNode* pRoot2){ if(pRoot2==NULL) return true; if(pRoot1==NULL) return false; if(pRoot1->val!=pRoot2->val) return false; return judge(pRoot1->left,pRoot2->left) && judge(pRoot1->right,pRoot2->right); } bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2) { bool res=false; if(pRoot1!=NULL&&pRoot2!=NULL){ if(pRoot1->val==pRoot2->val){ res=judge(pRoot1,pRoot2); } if(res==false){ res=HasSubtree(pRoot1->left,pRoot2); } if(res==false){ res=HasSubtree(pRoot1->right,pRoot2); } } return res; } };