一、理论基础
1、种类:满二叉树、完全二叉树、二叉搜索树、平衡二叉搜索树
2、存储方式:链式存储(左右结点指针和结点值)、顺序存储
3、遍历方式:深度优先(前序遍历、中序遍历、后序遍历)、广度优先(层次遍历)
二、递归遍历
三要素:
1、确定递归函数的参数和返回值
2、确定终止条件
3、确定单层递归的逻辑
三、迭代遍历
前序遍历:将父结点压入栈中,而后压入右子结点、左子结点,具体代码如下:
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;
}
};
中序遍历:由于访问和处理元素的顺序不一致,因此需要借助指针的帮助来访问结点,具体代码如下:
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;
}
};
后序遍历:与前序遍历类似,改变左右子结点入栈的顺序,最后将存储子结点值的数组进行反转即可。
四、统一迭代法
该方法将前、中、后序遍历的方法统一起来,核心思想在于将父结点入栈后出栈再入栈,第二次入栈时再入栈空结点进行标记,这样第二次出栈时可以此为标志进行出栈操作。下面给出中序遍历的代码,前序和后序遍历与之类似,只在入栈结点的顺序上做了改变。
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;
}
};