堆是什么?
堆被定义为具有特殊有序性质的树状结构。在最大堆中,对于任意父节点,其值大于或等于其子节点的值。在最小堆中,对于任意父节点,其值小于或等于其子节点的值。一般来讲,堆是一棵完全二叉树
在插入和删除元素时,堆具有O(logn)的时间复杂度,查找堆顶元素有O(1)的时间复杂度,非常快速
二叉树的遍历方式
二叉树主要有两种遍历方式:
- 深度优先遍历:先往深走,遇到叶子节点再往回走。根据中间节点的遍历顺序可以分为前序、中序、后序遍历三种遍历方式。
- 广度优先遍历:一层一层的去遍历。
//深度遍历的三种方式
#include <iostream>
// 二叉树的结点结构
struct TreeNode {
int value;
TreeNode* left;
TreeNode* right;
TreeNode(int val) : value(val), left(nullptr), right(nullptr) {}
};
// 前序遍历
void preorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
std::cout << root->value << " "; // 先访问根节点
preorderTraversal(root->left); // 递归遍历左子树
preorderTraversal(root->right); // 递归遍历右子树
}
// 中序遍历
void inorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
inorderTraversal(root->left); // 递归遍历左子树
std::cout << root->value << " "; // 先访问根节点
inorderTraversal(root->right); // 递归遍历右子树
}
// 后序遍历
void postorderTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
postorderTraversal(root->left); // 递归遍历左子树
postorderTraversal(root->right); // 递归遍历右子树
std::cout << root->value << " "; // 先访问根节点
}
int main() {
// 创建一棵二叉树
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
std::cout << "前序遍历结果:";
preorderTraversal(root);
std::cout << std::endl;
std::cout << "中序遍历结果:";
inorderTraversal(root);
std::cout << std::endl;
std::cout << "后序遍历结果:";
postorderTraversal(root);
std::cout << std::endl;
return 0;
}
//广度优先遍历
#include <iostream>
#include <queue>
struct TreeNode {
int value;
TreeNode* left;
TreeNode* right;
TreeNode(int val) : value(val), left(nullptr), right(nullptr) {}
};
void breadthFirstTraversal(TreeNode* root) {
if (root == nullptr) {
return;
}
std::queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
TreeNode* current = q.front();
q.pop();
std::cout << current->value << " ";
if (current->left) {
q.push(current->left);
}
if (current->right) {
q.push(current->right);
}
}
}
int main() {
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
std::cout << "广度优先遍历结果:";
breadthFirstTraversal(root);
std::cout << std::endl;
return 0;
}
二叉树的递归遍历
代码在上面已经写过了,三种不同的遍历方式没有本质上的区别
迭代遍历
迭代法使用循环结构(通常是栈或队列)来模拟递归过程,从而遍历二叉树,它的优点是可以有效地避免递归的深度过大导致的栈溢出问题,适用于大型二叉树,但是具有需要手动管理内存空间以及代码更加复杂的问题
和递归遍历相同的是迭代遍历也有三种方式,原则是什么序遍历就在处理对应节点的位置进行数据输出,最简单的是前序遍历,因为我们需要的是根节点并且需要输出的第一个节点也是根节点,代码如下:
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if (root == NULL) return result;
st.push(root);
while (!st.empty()) {
TreeNode* node = st.top(); // 中
st.pop();
result.push_back(node->val);
if (node->right) st.push(node->right); // 右(空节点不入栈)
if (node->left) st.push(node->left); // 左(空节点不入栈)
}
return result;
}
};
其中先入栈的是右节点,因为栈是FILO,左节点先出栈应该后入栈,对于后序遍历,只需要将“左右中”的顺序倒过来变成“中右左” ,在前序遍历“中左右”的基础上交换左右节点顺序再倒序输出即可
中序遍历相对复杂,因为此时元素的处理顺序和访问顺序不一致,此时使用指针辅助访问节点
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
stack<TreeNode*> st;
TreeNode* cur = root;
while (cur != NULL || !st.empty()) {
if (cur != NULL) { // 指针来访问节点,访问到最底层
st.push(cur); // 将访问的节点放进栈
cur = cur->left; // 左
} else {
cur = st.top(); // 从栈里弹出的数据,就是要处理的数据(放进result数组里的数据)
st.pop();
result.push_back(cur->val); // 中
cur = cur->right; // 右
}
}
return result;
}
};
统一迭代
统一迭代的代码比较复杂,但是相较上一种迭代的写法,更换代码顺序就可以写出三种遍历方式,编写的规则可以总结为在中节点访问后加入空节点做标记,从而改变总的访问顺序,只允许在访问到空节点后对应的数据出栈,。
以中序为例,从根节点开始向下访问到底部节点并按“右中左”的顺序入栈(反向出栈),其中所有的中节点前都插入一个NULL,也就是除了最底部的节点以外都按顺序排好了,并且都有NULL标记(相对于自己的左右节点而言大家都是中节点),对于底部的节点而言,因为没有字节点,自己最后也会被判定为一个没有左右节点的中节点,并且按照我们需要的顺序排列好了,剩下的工作就是出栈
//中序统一迭代遍历
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;
}
};