今日内容:
- 理论基础
- 递归遍历
- 迭代遍历
- 统一迭代
详细布置
理论基础
需要了解 二叉树的种类,存储方式,遍历方式 以及二叉树的定义
文章讲解:代码随想录
递归遍历 (必须掌握)
先序遍历
class Solution {
public:
void travelsal(TreeNode* cur, vector<int>& vec){
if(cur == NULL) return;//递归结束条件
vec.push_back(cur->val);//保存中间结点的值
travelsal(cur->left, vec);//左子树
travelsal(cur->right,vec);//右子树
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;//保存遍历结果值
travelsal(root, result);
return result;
}
};
中序遍历
class Solution {
public:
void travelsal(TreeNode* cur, vector<int>& val){
if(cur == NULL) return;
travelsal(cur->left, val);
val.push_back(cur->val);
travelsal(cur->right, val);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
travelsal(root,result);
return result;
}
};
后序遍历
class Solution {
public:
void traversal(TreeNode* cur, vector<int>& val){
if(cur == NULL) return;
traversal(cur->left, val);
traversal(cur->right, val);
val.push_back(cur->val);
}
vector<int> postorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root,result);
return result;
}
};
题目链接/文章讲解/视频讲解:代码随想录
迭代遍历 (基础不好的录友,迭代法可以放过)
先序遍历
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();//保存栈中结点指针
result.push_back(node->val);//将根节点或中间结点元素值入栈
st.pop();
if(node->right) st.push(node->right);
if(node->left) st.push(node->left);
}
return result;
}
};
后序遍历
入栈顺序 中 左 右
中结点入栈后立即出栈,左右孩子结点按栈,顺序弹出为右左
记录在数组中为中右左,最后翻转为左右中
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> result;
if(root == NULL) return result;
st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
result.push_back(node->val);
st.pop();
if(node->left) st.push(node->left);
if(node->right) st.push(node->right);
}
reverse(result.begin(),result.end());
return result;
}
};
中序遍历
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> result;
TreeNode* cur = root;
stack<TreeNode*> st;
while(cur != NULL || !st.empty()){
if(cur != NULL){
st.push(cur);
cur = cur->left;
}else{
cur = st.top();
st.pop();
result.push_back(cur->val);
cur = cur->right;
}
}
return result;
}
};
题目链接/文章讲解/视频讲解:代码随想录
统一迭代 (基础不好的录友,迭代法可以放过)
这是统一迭代法的写法, 如果学有余力,可以掌握一下
题目链接/文章讲解:代码随想录