【代码随想录笔记】二叉树(一)

while(77)

于 2024-08-10 07:55:26 发布

阅读量449

点赞数 10

文章标签：笔记

本文链接：https://blog.csdn.net/2301_80277275/article/details/140644698

版权

1、二叉树的递归遍历

递归遍历是一种深度优先的遍历方式，使用递归遍历二叉树时，需要想明白三个点：

1. 确定递归函数的参数和返回值

2. 确定终止条件

3. 确定单层递归的逻辑

编程语言使用递归是使用栈来实现的

这里我们使用leetCode上的前序、中序、后续遍历来作具体讲解

1. 递归函数的参数就是根节点和一个存放遍历结果的数组，返回值为void

2. 终止条件是当遍历到空时就返回

3. 单层递归逻辑就是先遍历当前结点，然后去遍历左子树，再去遍历右子树

class Solution {
public:
    void preorder(TreeNode* root,vector<int>& ret)
    {
        if(root == nullptr) return;
        ret.push_back(root->val);
        preorder(root->left,ret);
        preorder(root->right,ret);
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> ret;
        preorder(root,ret);
        return ret;
    }
};

1. 递归函数的参数就是根节点和一个存放遍历结果的数组，返回值为void

2. 终止条件是当遍历到空时就返回

3. 单层递归逻辑就是先遍历左子树，然后遍历当前结点，再去遍历右子树

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

1. 递归函数的参数就是根节点和一个存放遍历结果的数组，返回值为void

2. 终止条件是当遍历到空时就返回

3. 单层递归逻辑就是先遍历左子树，然后遍历右子树，再去遍历当前结点

class Solution {
public:
    void postorder(TreeNode* root,vector<int>& ret)
    {
        if(root == nullptr) return;
        postorder(root->left,ret);
        postorder(root->right,ret);
        ret.push_back(root->val);
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ret;
        postorder(root,ret);
        return ret;
    }
};

2、二叉树的非递归遍历

非递归遍历就是使用迭代法，编程语言能够实现递归是因为使用了栈，所以使用迭代法来模拟递归也是利用栈。理论上所有递归都可以使用栈来模拟，但是一些较为复杂的递归，使用栈模拟是十分困难的，也就没有必要使用栈来模拟了。所以，迭代法适用于较为简单的递归

2.1 前序遍历

首先创建一个存放结点指针的栈和一个存放遍历结果的数组，然后若是头结点为空，则直接返回，若是头结点不为空，则将头结点放入栈，进入循环，只要栈不为空就一直循环，每次循环，从栈中取出栈顶元素，并将这个结点的值放入数组中，并将这个结点不为空的孩子入栈。注意：前序遍历是先将右孩子入栈，再将左孩子入栈，因为前序遍历是中左右，而栈是后进先出，所以先将右孩子入栈，再将左孩子入栈，是可以先遍历到左孩子的，也就符合中左右

以这颗二叉树为例

正式因为孩子中空结点是不会放入的，所以在前面将头结点放入后，一定要检查一下头结点是否为空，若是空，后面解引用会报错

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> ret;
        if (root == nullptr) return ret; // 这里一定要判断一下root是否为空，因为后面都是不为空的结点才放入栈，若这里是空，则后面对其解引用会出错
        st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();                       // 中
            st.pop();
            ret.push_back(node->val);
            if (node->right) st.push(node->right);           // 右（空节点不入栈）
            if (node->left) st.push(node->left);             // 左（空节点不入栈）
        }
        return ret;
    }
};

2.2 后序遍历

前序遍历是中左右，后序遍历是左右中，所以只需要将前序遍历的迭代法中左孩子和右孩子入栈的顺序交换一下，并将结果数组逆置一下就是后序遍历

中左右 -> 中右左 -> 左右中

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> ret;
        if(root == nullptr) return ret;
        st.push(root);
        while(!st.empty())
        {
            TreeNode* node = st.top();
            st.pop();
            ret.push_back(node->val);
            if(node->left) st.push(node->left);
            if(node->right) st.push(node->right);
        }
        reverse(ret.begin(),ret.end());
        return ret;
    }
};

2.3 中序遍历

中序遍历与前序有一些区别，所以无法通过上面代码改一下顺序变成中序。前序遍历到的结点和要处理的结点是一个顺序，而中序遍历遍历到的结点和要处理(将结点放进数组)的结点不是一个顺序。以上面那颗二叉树为例，前序从5开始遍历，5同时也是要处理的结点，然后到4，4也是要处理的结点，而中序也是从5开始，但5并不是要处理的结点，还需要一直向左，直到到2，2才是第一个要处理的结点，但2已经是遍历到的第三个结点了。后序遍历到的结点和要处理的结点不是一个顺序，只是因为其比较特殊，能通过与前序的关联改几行代码而实现，但是中序不行。

中序遍历的迭代法需要用一个指针cur来遍历二叉树，仍然需要一个栈和一个存放结果的数组。只有当栈为空并且cur也为空才会结束循环。这里的栈是存放遍历到的结点的指针，此时遍历到一个结点，并不会将其放入数组，只是放入栈。最先开始，让cur指向根节点，然后一直向左一直向左，直到cur为空，此时栈顶的那个结点就是这颗树最左边的结点(注意，不一定是叶子结点)，让cur指向这个结点，然后将它的值放入数组，并pop掉，现在是放入了中，然后就是右，让cur变成cur的right，因为cur的左孩子为空，但是右孩子不一定为空，操作与上面是一样的。总而言之，就是只要cur为空，就取栈顶的结点的值放进数组，然后cur指向这个结点的右孩子，然后pop,cur不为空，就将这个结点的指针放入栈，然后cur指向cur的left

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> ret;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while(cur || !st.empty())
        { // 指针访问结点，访问到最底层
            if(cur != nullptr) 
            {
                st.push(cur); // 将访问到的结点放入栈
                cur = cur->left; // 左
            }
            else
            {
                cur = st.top(); // 从栈里弹出的数据，就是要处理的数据（放进ret数组里的数据）
                st.pop();
                ret.push_back(cur->val); // 中
                cur = cur->right; // 右
            }
        }
        return ret;
    }
};