算法小白训练营日记,笔记为自用,若有错误感谢指出
--今日任务--:二叉树的递归遍历,二叉树的迭代遍历,二叉树的层序遍历
链式存储的二叉树节点的定义方式:
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
二叉树的递归遍历
前序遍历:中左右;中序遍历:左中右;后序遍历:左右中。
代码如下:
class Solution {
public:
//前序遍历
void traversalq(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
vec.push_back(cur->val); // 中
traversal(cur->left, vec); // 左
traversal(cur->right, vec); // 右
}
//中序遍历
void traversalz(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
traversal(cur->left, vec); // 左
vec.push_back(cur->val); // 中
traversal(cur->right, vec); // 右
}
//后序遍历
void traversalh(TreeNode* cur, vector<int>& vec) {
if (cur == NULL) return;
traversal(cur->left, vec); // 左
traversal(cur->right, vec); // 右
vec.push_back(cur->val); // 中
}
vector<int> preorderTraversal(TreeNode* root) {
vector<int> result;
traversalq(root, result);
return result;
}
};
二叉树的迭代遍历
前序遍历:
运用栈来完成操作,将根节点压入栈中,node指针指向栈顶元素,弹出栈顶元素即根节点,遍历node指针的左右子树,因为栈为先入后出,所以为了得到结果后先将左子树的元素弹出,应该先将右子树压入栈。
具体代码如下:
class Solution {
public:
vector<int> preorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
if(root==NULL) return res;
st.push(root);
while(!st.empty()){
TreeNode* node=st.top();
st.pop();
res.push_back(node->val);
if(node->right) st.push(node->right);
if(node->left) st.push(node->left);
}
return res;
}
};
后序遍历:前序遍历为中左右,将左右子树压入栈位置对调后得到遍历结果为中右左,倒置数组得到左右中即为后序遍历。
具体代码如下:
class Solution {
public:
vector<int> postorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
if(root==nullptr) return res;
st.push(root);
while(!st.empty()){
TreeNode* node=st.top();
st.pop();
res.push_back(node->val);
if(node->left) st.push(node->left);
if(node->right) st.push(node->right);
}
reverse(res.begin(),res.end());
return res;
}
};
中序遍历:若cur指针指向节点不为空时,一直遍历左子树,将遍历过的节点压入栈中否则将栈顶元素弹出栈,并遍历栈顶元素的右子树。
具体代码如下:
class Solution {
public:
vector<int> inorderTraversal(TreeNode* root) {
stack<TreeNode*> st;
vector<int> res;
if(root==nullptr) return res;
st.push(root);
TreeNode* node=root;
while(cur!=NULL||!st.empty()){
if(node!=NULL){
st.push(node);
node=node->left;
}else{
node=st.top();
st.pop();
res.push_back(node->val);
node=node->right;
}
}
return res;
}
};
二叉树的层序遍历
层序遍历用队列结构解决,先将根节点存入队列中,队头出一个元素,队尾进一个元素的左右节点,每遍历完一层,记录该层节点数size,每次将队列中前size个节点存入到数组中。
具体代码如下:
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
queue<TreeNode*> que;
if(root!=NULL) que.push(root);
while(!que.empty()){
int size=que.size();
vector<int> vec;
for(int i=0;i<size;i++){
TreeNode* node=que.front();
que.pop();
vec.push_back(node->val);
if(node->left) que.push(node->left);
if(node->right) que.push(node->right);
}
res.push_back(vec);
}
return res;
}
};