学习文章链接:代码随想录
熟悉二叉树的定义
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
一、递归遍历
题目链接:前序递归遍历中左右
class Solution {
public:
void traversal(TreeNode *tree,vector<int> &vec){
if(tree==nullptr){
return;
}
vec.push_back(tree->val);
traversal(tree->left,vec);
traversal(tree->right,vec);
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
traversal(root,result);
return result;
}
};
二、迭代遍历
题目链接:同上
思路:使用栈的方法,中序和前序有很大区别
需要注意的点:后序遍历的结果需要反转一下
后序遍历:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
vector<int> results;
stack<TreeNode*> st;
if(root==nullptr)return results;
st.push(root);
while(!st.empty()){
TreeNode* node=st.top();
st.pop();
if(node->left){st.push(node->left);}
if(node->right){st.push(node->right);}
results.push_back(node->val);
}
// reverse(results.begin(), results.end());
for(int i=0,j=results.size()-1;i<results.size()/2;i++,j--){
int cache=results[i];
results[i]=results[j];
results[j]=cache;
}
return results;
}
};
中序遍历:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
vector<int> results;
stack<TreeNode*> st;
if(root==nullptr)return results;
TreeNode* cur=root;
while(!st.empty()||cur!=nullptr){
if(cur!=nullptr){
st.push(cur);
cur=cur->left;
}
else{
cur=st.top();
st.pop();
results.push_back(cur->val);
cur=cur->right;
}
}
return results;
}
};
三、统一迭代
题目链接:同上
思路:代码随想录二叉树的统一迭代
没看懂,先记录一下
四、层序遍历
一口气刷十题
102.二叉树的层序遍历
107.二叉树的层次遍历II
199.二叉树的右视图
637.二叉树的层平均值
429.N叉树的层序遍历
515.在每个树行中找最大值
116.填充每个节点的下一个右侧节点指针
117.填充每个节点的下一个右侧节点指针II
104.二叉树的最大深度
104.二叉树的最大深度
代码模板:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> que;
vector<vector<int>> results;
if(root==nullptr)return results;
que.push(root);
while(!que.empty()){
int size=que.size();
vector<int> result;
for(int i=0;i<size;i++){
TreeNode* node=que.front();
result.push_back(node->val);
que.pop();
if(node->left)que.push(node->left);
if(node->right)que.push(node->right);
}
results.push_back(result);
}
return results;
}
};