算法训练营 第十四天
二叉树 day01~
二叉树的基础知识
二叉树的种类
- 满二叉树:这个树的结点都是满的
- 完全二叉树: 这个树的结点是顺序长的,不能中间跳过几个再长结点。
- 二叉搜索树:和满二叉树以及完全二叉树不同,这类树指的是结点值都是有顺序的,是用来搜索的,对于树的结点结构没有要求。
- 平衡二叉搜索树:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树,比如红黑树就是平衡二叉树。
二叉树的存储方式
- 链式存储:也就是用链表存储二叉树
- 顺序存储:用数组存储二叉树,一个下标为i的节点,它的左子结点的下标为2i+1,右子结点的下标为2i+2
二叉树的遍历
- 深度优先搜索(递归,用栈实现递归的行为;迭代法)
1、前序遍历:访问节点的顺序——中左右
2、中序遍历:访问节点的顺序——左中右
3、后序遍历:访问节点的顺序——左右中 - 广度优先搜索(层序遍历,迭代法)
递归遍历
写递归算法有三要素:
1、确定递归函数的参数和返回值
2、确定终止条件
3、确定单层递归的逻辑
按照这三要素来写就能完成递归算法了。
迭代遍历
迭代法是用栈来模拟递归的过程
前后序遍历
前序遍历如下,后序遍历是将插入栈的顺序改为中右左,处理完栈后反转数组得到的
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
auto cur = root;
if (root==nullptr)
return res;
st.push(root);
while (!st.empty()) {
auto tmp = st.top();
res.push_back(tmp->val);
st.pop();
if(tmp->right!=nullptr)
st.push(tmp->right);
if(tmp->left!=nullptr)
st.push(tmp->left);
}
return res;
}
};
中序遍历
中序遍历的逻辑是左中右,先将结点及其左节点插入栈中,在弹出结点的时候处理它的右节点
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
auto cur = root;
while (cur != nullptr) {
st.push(cur);
cur = cur->left;
}
while (!st.empty()) {
auto tmp = st.top();
res.push_back(tmp->val);
st.pop();
tmp = tmp->right;
while (tmp != nullptr) {
st.push(tmp);
tmp = tmp->left;
}
}
return res;
}
};
统一迭代
统一迭代将前中后序遍历的代码统一了一下,实现逻辑是,顺序访问结点,但压入栈的顺序不一样。
- 前序统一迭代
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop();
if (node->right) st.push(node->right); // 右
if (node->left) st.push(node->left); // 左
st.push(node); // 中
st.push(NULL);
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};
- 中序统一迭代
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop(); // 将该节点弹出,避免重复操作,下面再将右中左节点添加到栈中
if (node->right) st.push(node->right); // 添加右节点(空节点不入栈)
st.push(node); // 添加中节点
st.push(NULL); // 中节点访问过,但是还没有处理,加入空节点做为标记。
if (node->left) st.push(node->left); // 添加左节点(空节点不入栈)
} else { // 只有遇到空节点的时候,才将下一个节点放进结果集
st.pop(); // 将空节点弹出
node = st.top(); // 重新取出栈中元素
st.pop();
result.push_back(node->val); // 加入到结果集
}
}
return result;
}
};
- 后序统一迭代
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
if (root != NULL) st.push(root);
while (!st.empty()) {
TreeNode* node = st.top();
if (node != NULL) {
st.pop();
st.push(node); // 中
st.push(NULL);
if (node->right) st.push(node->right); // 右
if (node->left) st.push(node->left); // 左
} else {
st.pop();
node = st.top();
st.pop();
result.push_back(node->val);
}
}
return result;
}
};