Day14- 第六章 二叉树part01
今日内容:
● 理论基础
● 递归遍历
● 迭代遍历
● 统一迭代
今日知识点:
● 优先队列
144. 二叉树的前序遍历
1.解题思路
遍历值引用数组地址nums,递归参数为根节点和数组,终止条件是根节点为空,单层就是前序为中左右,后序为左右中,中序为左中右。
2.代码
- 递归法
class Solution {
public:
void pre(TreeNode* root,vector<int>& nums){
if(root == nullptr)
return;
nums.push_back(root->val);
pre(root->left,nums);
pre(root->right,nums);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> ans;
pre(root,ans);
return ans;
}
};
- 迭代法
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> ans;
if(root == nullptr) return ans;
st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
st.pop();
if(node != nullptr)
ans.push_back(node->val);
else
continue;
st.push(node->right);
st.push(node->left);
}
return ans;
}
};
145. 二叉树的后序遍历
2.代码
- 递归法
class Solution {
public:
void post(TreeNode* root,vector<int>& res){
if(root == nullptr)
return;
post(root->left,res);
post(root->right,res);
res.push_back(root->val);
}
vector<int> postorderTraversal(TreeNode* root) {
vector<int> ans;
post(root,ans);
return ans;
}
};
- 迭代法:前序遍历为中左右,后序遍历为左右中,那么前序遍历为中右左,数组反转即为左右中
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> ans;
stack<TreeNode*> st;
if(root == nullptr) return ans;
st.push(root);
while(!st.empty()){
TreeNode* node = st.top();
st.pop();
ans.push_back(node->val);
if(node->left) st.push(node->left);
if(node->right) st.push(node->right);
}
reverse(ans.begin(),ans.end());
return ans;
}
};
94. 二叉树的中序遍历
2.代码
class Solution {
public:
void Inorder(TreeNode* root,vector<int>& ans){
if(root == nullptr)
return;
Inorder(root->left,ans);
ans.push_back(root->val);
Inorder(root->right,ans);
}
vector<int> inorderTraversal(TreeNode* root) {
vector<int> ans;
Inorder(root,ans);
return ans;
}
};
- 迭代法 一直往左遍历到空,取出根结底,遍历根节点右。
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> ans;
TreeNode* cur = root;
stack<TreeNode*> st;
while(!st.empty() || cur != nullptr){
if(cur){
st.push(cur);
cur = cur->left;
}
else{
cur = st.top();
ans.push_back(cur->val);
st.pop();
cur = cur->right;
}
}
return ans;
}
};
总结
- 递归法
递归法遍历使用数组记录遍历顺序,前序push,left,right;中序left,push,right,后序left,right,push; - 迭代法
迭代法使用栈的特点,前序遍历一个节点就push,判断左右是否为空,不为空放入栈(先进后出,所以先右后左儿子);后序遍历为中右左,反转数组为中左右;中序遍历因为push和遍历不是同步的,故用额外指针cur遍历,遍历的节点都存入栈,一直遍历左儿子,节点为空时,取出栈头(根)节点push,遍历右儿子…直到栈为空而且cur也空才停止。