算法训练营 第十五天
二叉树 day02~
层序遍历
层序遍历的关键在于,他不能再用栈来处理了,得用队列,每层有几个结点,用一个变量保存。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root == nullptr) {
return { {} };
}
vector<vector<int>> res;
queue<TreeNode*> que;
que.push(root);
while (!que.empty()) {
vector<int> Lev;
int size = que.size();
while (size--) {
auto tmp = que.front();
Lev.push_back(tmp->val);
que.pop();
if (tmp->left != nullptr) que.push(tmp->left);
if (tmp->right != nullptr) que.push(tmp->right);
}
res.push_back(Lev);
}
return res;
}
};
反转二叉树
反转二叉树,用前序和后序遍历都行,中序遍历要注意,会有左节点被重复操作的情况,改成左中左的遍历顺序也可以实现(先处理左节点,处理过后左节点变为了右结点,所以处理完中结点后要再处理左节点)
class Solution {
public:
TreeNode* invertTree(TreeNode* root) {
if (root == NULL) return root;
swap(root->left, root->right); // 中
invertTree(root->left); // 左
invertTree(root->right); // 右
return root;
}
};
对称二叉树
对称二叉树,用递归的写法,归根结底是比较左节点和右节点是否是互为反转二叉树,将根节点的左子树反转之后再和右子树比较也行,但是会浪费大量的时间,更科学的比较方式是一边遍历一边比较。用递归的三步法
1、先确定输入输出,输入左右节点,返回bool类型值
2、再确定停止条件,如果左右结点都为空,返回true;如果左右节点有且仅有一个为空,返回false;如果左右节点的val不相等,返回false;剩下的情况就是左右节点都不为空且值相等的情况了,这种情况就继续往下递归。
3、确定单次递归逻辑,分别比较左右结点的外侧结点和内侧结点是否相同,如果都相同则返回true,否则返回false;
class Solution {
public:
bool digui(TreeNode* left, TreeNode* right){
if(left==nullptr&&right==nullptr) return true;
else if(left==nullptr&&right!=nullptr) return false;
else if(left!=nullptr&&right==nullptr) return false;
else if(left->val!= right->val) return false;
bool outside = digui(left->left,right->right);
bool inside = digui(left->right,right->left);
return (outside&&inside);
}
bool isSymmetric(TreeNode* root) {
if(root==nullptr)
return true;
return digui(root->left,root->right);
}
};