Day 7
26. 树的子结构
输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
解题思路
若树B是树A的子结构,则子结构的根节点可能为树A的任意一个节点。因此,判断树B是否是树A的子结构,需完成以下两步工作:
1.查找A中与B根节点相同的节点;
2.判断A的与B根节点相同的节点的左右子树是否相同。
class Solution {
public:
bool isSubStructure(TreeNode* A, TreeNode* B) {
if(A == NULL || B == NULL) return false;
if(A->val == B->val && helper(A->left, B->left) && helper(A->right, B->right))
{
return true;
}
return isSubStructure(A->left, B) || isSubStructure(A->right, B);
}
private:
bool helper(TreeNode* A, TreeNode* B)
{
if(B == NULL) return true;
if(A == NULL) return false;
if(A->val == B->val && helper(A->left, B->left) && helper(A->right, B->right))
{
return true;
}
return false;
}
};
27. 二叉树的镜像
请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
4
/ \
2 7
/ \ / \
1 3 6 9
镜像输出:
4
/ \
7 2
/ \ / \
9 6 3 1
示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
解题思路
方法一:递归
这是一道很经典的二叉树问题。显然,我们从根节点开始,递归地对树进行遍历,并从叶子节点先开始翻转得到镜像。如果当前遍历到的节点的左右两棵子树都已经翻转得到镜像,那么我们只需要交换两棵子树的位置,即可得到以root 为根节点的整棵子树的镜像。
递归翻转子树
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root) {
if(root == NULL) return root;
TreeNode* temp = root->left;
root->left = root->right;
root->right = temp;
mirrorTree(root->left);
mirrorTree(root->right);
return root;
}
};
方法二:辅助栈(或队列)
利用栈(或队列)遍历树的所有节点node ,并交换每个node 的左/右子节点。
算法流程:
特例处理:当root为空时,直接返回null;
初始化:栈(或队列),本文用栈,并加入根节点 rootroot 。
循环交换: 当栈stack 为空时跳出;
出栈:记为 nodenode;
添加子节点:将node左和右子节点入栈;
交换:交换node的左/右子节点。
返回值:返回根节点root。
class Solution {
public:
TreeNode* mirrorTree(TreeNode* root) {
if(root == nullptr) return nullptr;
stack<TreeNode*> stack;
stack.push(root);
while (!stack.empty())
{
TreeNode* node = stack.top();
stack.pop();
if (node->left != nullptr) stack.push(node->left);
if (node->right != nullptr) stack.push(node->right);
TreeNode* tmp = node->left;
node->left = node->right;
node->right = tmp;
}
return root;
}
};
28. 对称的二叉树
请实现一个函数,用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样,那么它是对称的。
例如,二叉树 [1,2,2,3,4,4,3] 是对称的。
1
/ \
2 2
/ \ / \
3 4 4 3
但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:
1
/ \
2 2
\ \
3 3
解题思路
递归判断:
1.树的的两个根结点具有相同的值
2.每个树的右子树都与另一个树的左子树镜像对称
class Solution {
public:
bool isSymmetric(TreeNode* root) {
return check(root, root);
}
private:
bool check(TreeNode* p, TreeNode* q) {
if (!p && !q) return true;
if (!p || !q) return false;
return p->val == q->val && check(p->left, q->right) && check(p->right, q->left);
}
};