代码随想录算法训练营第十四天|LeetCode144 二叉树的前序遍历、LeetCode 145 二叉树的后序遍历、LeetCode95 二叉树的中序遍历

前序遍历:中左右

后序遍历:左右中

中序遍历:左中右

​​​​​144. 二叉树的前序遍历

思路:1.递归遍历,构造递归函数,按照中左右的顺序遍历二叉树的所有元素放入数组中。

/**
 * 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:
    //递归函数
    void traversal(TreeNode* cur,vector<int>& result)
    {
        if(cur==NULL)
        {
            return;
        }
        result.push_back(cur->val);
        traversal(cur->left,result);
        traversal(cur->right,result);
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root,result);
        return result;
    }
};

2、非递归遍历:利用迭代遍历、利用栈先入后出的特点,遍历二叉树的元素放入栈中,先放入根节点,然后弹出栈的头元素(根节点),然后将根节点的右节点和左节点输入栈中,这样的话栈的头部就是左节点,弹出的时候先弹出左节点,只要栈不为空就不结束循环,下个循环中头节点为左节点,此时再弹出它,在判断它的左右节点。这样就能保证树的所有元素都是按前序(左中右)顺序进入的数组。

class Solution {
public:
    //利用迭代遍历 栈
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> result;
        if (root==NULL)
        {
            return result;
        }
        else
        {
            st.push(root);
        }
        while(!st.empty())
        {
            TreeNode* node = st.top();
            st.pop();
            result.push_back(node->val);
            if(node->right!=NULL)
            {
                st.push(node->right);
            }
            if(node->left!=NULL)
            {
                st.push(node->left);
            }
        }
        return result;
    }
};

145.二叉树的后序遍历

思路:1、递归遍历,跟前序遍历的原理相同,只需要在递归函数中改变顺序。

/**
 * 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:
    void traversal(TreeNode* cur,vector<int>& result)
    {
        if(cur==NULL)
        {
            return;
        }
        traversal(cur->left,result);
        traversal(cur->right,result);
        result.push_back(cur->val);
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root,result);
        return result;
    }
};

2、非递归遍历:迭代遍历:原理同前序遍历,后序遍历的顺序为左右中,因此我们将前序遍历的循环中,左右节点入栈的顺序变化,此时前序的顺序变为中右左, 循环结束后,再将数组翻转,即为左右中顺序。

class Solution {
public:
    //迭代遍历 栈
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;

        vector<int> result;
        if(root==NULL)
        {
            return result;
        }
        else
        {
            st.push(root);
        }
        while(!st.empty())
        {
            TreeNode*node = st.top();
            st.pop();
            result.push_back(node->val);
            if(node->left!=NULL)
            {
                st.push(node->left);
            }
            if(node->right!=NULL)
            {
                st.push(node->right);
            }
        }
        reverse(result.begin(),result.end());
        return result;
    }
};

94.二叉树的中序遍历

思路:1、递归遍历:原理同前序、后序,秩序将递归函数中的顺序改为左中右。

class Solution {
public:
    void traversal(TreeNode* cur,vector<int>& result)
    {
        if(cur==NULL)
        {
            return;
        }
        traversal(cur->left,result);
        result.push_back(cur->val);
        traversal(cur->right,result);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root,result);
        return result;
    }
};

2、非递归遍历:迭代遍历,先将头节点放入栈中,然后定义一个指针指向根节点,只要根节点不为空,或者栈不为空,循环就不停止,循环内部,如果指针不为空,就把指针处的节点放入栈中,然后指针指向节点左下节点,找到树的最左节点。当指针指向空时,说明上一个节点没有左节点,将指针指向上一个节点,即栈顶,然后把栈顶元素弹出,把这个元素的值输入数组,然后再判断这个节点的右下节点,如果没有右下节点,继续将指针指向栈顶元素,重复上述循环。如果有右下节点,将这个节点放入栈中,找这个节点是否有左下节点,重复循环。

class Solution {
public:
    //迭代遍历
    vector<int> inorderTraversal(TreeNode* root) {
        //定义一个数组用于记录结果 一个栈用来记录遍历的元素
        stack<TreeNode*> st;
        TreeNode* cur = root;
        vector<int> result;
        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;
    }
};

收获:

1、开始补春节欠的债,第一天接触二叉树,学会了二叉树的基本概念。

2、递归遍历中只需要改动递归函数中的顺序,即可实现前序、后序、中序的遍历。

  • 9
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值