本文主要是介绍剑指offer--树的子结构(牛客网),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
题目描述
输入两棵二叉树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){if(pRoot2 == nullptr) return false;if(pRoot1 == nullptr) return false;bool flag = isSubtree(pRoot1, pRoot2);if(flag == false)return HasSubtree(pRoot1->left,pRoot2) || HasSubtree(pRoot1->right, pRoot2);else return true;}bool isSubtree(TreeNode* p1, TreeNode* p2){if(p2 == nullptr) return true;if(p1 == nullptr) return false;if(p1->val == p2->val){return isSubtree(p1->left, p2->left) && isSubtree(p1->right, p2->right);}elsereturn false;}
};
需要注意两点:
1)保证B为空时,返回false
在 HasSubtree 中判断 pRoot2 为空时, 返回false.
而在 isSubtree 中,已经确定 根节点 pRoot2 是非空的,所以后续子节点如果为空,则到达了叶节点,返回true.
2)B为A的子结构不一定在B中的叶节点在A中也全是叶节点。如 [8,8,7,9,2,#,#,#,#,4,7] 与[8,9,2]
所以在isSubtree中判断为"p1->val == p2->val",而不能为"p1 == p2"(包含了val, left 和 right).
这篇关于剑指offer--树的子结构(牛客网)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!