代码随想录跟练日记|Day13 二叉树part01

算法小白训练营日记，笔记为自用，若有错误感谢指出

--今日任务--：二叉树的递归遍历，二叉树的迭代遍历，二叉树的层序遍历

链式存储的二叉树节点的定义方式：

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;
    }
};