1. 题目来源
链接:572. 另一棵树的子树
2. 题目解析
看到这个题目就感觉不简单,因为写了写 dfs 版本的,发现好像不太会…
还是简单粗暴一点,直接搞一个 前序+中序,进行判断即可。我们知道通过 前序+中序,是可以构建出一颗唯一的二叉树的,当然可以通过 前序+中序,去判断两颗二叉树是不是一样的。
但这里需要注意的是:
- 我们需要将 NULL 的位置也通过占位标记记录一下,不然无法判断子树完全相等,只能判断出来存在相同的子结构。例如:
但这里需要判断两个 vector a、b,b 是否在 a 中出现…这个东西写起来比较耗性能。但在这还是过了。算是一个思路吧。
dfs:
每个点,都可能是目标子树的根节点,同时我们需要判断当根节点确定时,该根节点的子树是否等于目标子树。故需要两个递归函数:
- dfs 函数:遍历树上所有节点,判断以该节点作为根节点,它的子树是否有包含目标子树。
- 先判断当前根节点是否可以作为目标子树的根节点。
- 再判断根节点的左子树、右子树下的所有节点是否可以作为目标子树的根节点。
- check 函数:判断节点所处子树是否等于目标子树。
至于其他的写法,看官解吧。还是很秀的…
评论区有提到 树上 HASH 的方法字节面试过…让写一下…
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( 1 ) O(1) O(1)
前、中 序判断二叉树。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> a1, a2, b1, b2;
void dfs1(TreeNode* root, vector<int> &v) {
if (!root) {v.push_back(1e9); return;}
v.push_back(root->val);
dfs1(root->left, v);
dfs1(root->right, v);
}
void dfs2(TreeNode* root, vector<int> &v) {
if (!root) {v.push_back(1e9); return;}
dfs2(root->left, v);
v.push_back(root->val);
dfs2(root->right, v);
}
bool check(vector<int> &a, vector<int>& b) {
int n = a.size(), m = b.size();
for (int i = 0; i < n; i ++ ) {
bool flag = false;
for (int k = i, j = 0; k < n && j < m; j ++ , k ++ ) {
if (a[k] != b[j]) {
break;
}
if (j == m - 1) flag = true;
}
if (flag) return true;
}
return false;
}
bool isSubtree(TreeNode* root, TreeNode* subRoot) {
dfs1(root, a1);
dfs2(root, a2);
dfs1(subRoot, b1);
dfs2(subRoot, b2);
return check(a1, b1) && check(a2, b2);
}
};
dfs:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool check(TreeNode* root, TreeNode* subRoot) {
if (!root && !subRoot) return true;
if (!root && subRoot) return false;
if (root && !subRoot) return false;
if (root->val != subRoot->val) return false;
// 同步判断子结构是否一致
return check(root->left, subRoot->left) && check(root->right, subRoot->right);
}
// 判断树 root 下是否存在 subRoot 结构的子树
bool dfs(TreeNode* root, TreeNode* subRoot) {
if (!root) return false;
// 先判断root根是否可以作为目标子树的根
// root根无法作为目标子树根,目标子树根可能存在root的左子树、右子树当中
// dfs 判断左子树 是否存在目标子树
// dfs 判断右子树 是否存在目标子树
return check(root, subRoot) || dfs(root->left, subRoot) || dfs(root->right, subRoot);
}
bool isSubtree(TreeNode* root, TreeNode* subRoot) {
return dfs(root, subRoot);
}
};