二叉树的中序遍历

WJX5742024-08-10 14:04

法一：递归，最简单的方法

直接上代码：注意递归终止条件为碰到空结点。

cpp 复制代码

/**
 * 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 {
    vector<int>ans;
    void zxbl(TreeNode* node)
    {
        if(!node) return;//注意递归都要有终止条件，这里递归终止的条件为碰到空节点。
        zxbl(node->left);
        ans.push_back(node->val);//加入当前结点
        zxbl(node->right);
    }
public:
    vector<int> inorderTraversal(TreeNode* root) {
        zxbl(root);
        return ans;
    }
};

法二：迭代（要掌握！）

递归函数我们也可以用迭代的方式实现，两种方式是等价的，区别在于递归的时候隐式地维护了一个栈，而我们在迭代的时候需要显式地将这个栈模拟出来，其他都相同，具体实现可以看下面的代码。

关键思路就是这个栈（先入后出）！首先对于每个结点就先放入栈中，然后看它有没有左孩子，有左孩子就将左孩子继续放入栈中，没有左孩子就把当前结点输出，再看它有没有右孩子，有的话就放入栈中，继续上述操作，每次都处理当前结点。

代码其实之前有写过类似的。

注意：栈和队列是push，vector是push_back

代码：

cpp 复制代码

/**
 * 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> ans;
        stack<TreeNode *> st;//栈，存每一个结点，是指针类型的
        while(root||!st.empty())//栈不空
        {
            while(root)
            {
                st.push(root);
                root=root->left;
            }
            auto t=st.top();
            st.pop();
            ans.push_back(t->val);
            root=t->right;

        }
        return ans;
    }
};

这个迭代的代码需要好好理解，为什么这么写，好好思考琢磨。