C语言实现二叉树的中序遍历

二叉树是一种重要的数据结构，对二叉树的遍历也很重要。这里简单介绍三种二叉树中序遍历的方法。二叉树的中序遍历就是首先遍历左子树，然后访问当前节点，最后遍历右子树。对于下面的二叉树，中序遍历结果如下：

结果：[5,10,6,15,2]

直观来看，二叉树的中序遍历就是将节点投影到一条水平的坐标上。如图：

1、递归法

这是思路最简单的方法，容易想到并且容易实现。递归的终止条件是当前节点是否为空。首先递归调用遍历左子树，然后访问当前节点，最后递归调用右子树。代码如下：

void inorder(struct TreeNode* root, int* res, int* resSize) {
    if (!root) {
        return;
    }
    //递归到最后一定是一个左右子树都为空的节点
    inorder(root->left, res, resSize);//先左
    res[(*resSize)++] = root->val;//然后中间节点
    inorder(root->right, res, resSize);//最后右边
}

int* inorderTraversal(struct TreeNode* root, int* returnSize) {
    int* res = malloc(sizeof(int) * 501);
    *returnSize = 0;
    inorder(root, res, returnSize);
    return res;
}

复杂度分析

时间复杂度：O(n)，其中 nn 为二叉树节点的个数。二叉树的遍历中每个节点会被访问一次且只会被访问一次。
空间复杂度：O(n)。空间复杂度取决于递归的栈深度，而栈深度在二叉树为一条链的情况下会达到 O(n) 的级别。

2、迭代法-栈

在迭代方法中，从根节点开始找二叉树的最左节点，将走过的节点保存在一个栈中，找到最左节点后访问，对于每个节点来说，它都是以自己为根的子树的根节点，访问完之后就可以转到右子上了。代码如下：

int* inorderTraversal(struct TreeNode* root, int* returnSize) {
	//初始化
    *returnSize = 0;
    //为返回的的数组分配空间
    int* res = malloc(sizeof(int) * 501);
    //创建一个存储TreeNode指针的指针数组，也就是栈
    struct TreeNode** stk = malloc(sizeof(struct TreeNode*) * 501);
    int top = 0;//栈的指示位，一开始指向0
    //最后跳出大循环的时候，root==NULL，top == 0
    while (root != NULL || top > 0) {
    	//压栈操作，将当前节点以及所有其左子树全部压入
        while (root != NULL) {
            stk[top++] = root;
            root = root->left;
        }
        //将栈顶的数据弹出
        root = stk[--top];
        res[(*returnSize)++] = root->val;
        
        root = root->right;
    }
    return res;
}

复杂度分析

时间复杂度：O(n)，其中 nn 为二叉树节点的个数。二叉树的遍历中每个节点会被访问一次且只会被访问一次。
空间复杂度：O(n)。空间复杂度取决于递归的栈深度，而栈深度在二叉树为一条链的情况下会达到 O(n) 的级别。

3、Morris法

这种方法是Morris发明的，看完之后感觉精妙无比。这种方法不使用递归，不使用栈，O(1)的空间复杂度完成二叉树的遍历。这种方法的基本思路就是将所有右儿子为NULL的节点的右儿子指向后继节点（对于右儿子不为空的节点，右儿子就是接下来要访问的节点）。这样，对于任意一个节点，当访问完它后，它的右儿子已经指向了下一个该访问的节点。对于最右节点，不需要进行这样的操作。注意，这样的操作是在遍历的时候完成的，完成访问节点后会把树还原。整个循环的判断条件为当前节点是否为空。例如上面的二叉树，遍历过程如下（根据当前节点c的位置）：

（1）当前节点为10，因为左儿子非空，不能访问，找到c的左子树的最右节点p：

结果：[]

（2）找节点c的左子树的最右节点有两种终止条件，一种右儿子为空，一种右儿子指向当前节点。下面是右儿子为空的情况，这种情况先要构造，将节点p的右儿子指向后继节点c，然后c下移：

结果：[]

（3）当前节点c的左儿子为空，进行访问。访问后将c指向右儿子（即后继节点）：

结果：[5]

（4）继续寻找左子树的最右节点，这次的终止条件是最右节点为当前节点。这说明当前节点的左子树遍历完毕，访问当前节点后，还原二叉树，将当前节点指向后继节点：

结果：[5,10]

（5）重复上述过程，直到c指向整棵二叉树的最右节点：

左儿子为空，进行访问，c转到右儿子。右儿子为空，不满足判断条件，循环结束，遍历完成。结果如下：

[5,10,6,15,2]

这就是Morris方法，时间复杂度为O(n)，空间复杂度是O(1)。代码如下：

int* inorderTraversal(struct TreeNode* root, int* returnSize) {
    int* res = malloc(sizeof(int) * 501);
    *returnSize = 0;
    struct TreeNode* predecessor = NULL;

    while (root != NULL) {
        if (root->left != NULL) {
            // predecessor 节点就是当前 root 节点向左走一步，然后一直向右走至无法走为止
            predecessor = root->left;
            while (predecessor->right != NULL && predecessor->right != root) {
                predecessor = predecessor->right;
            }

            // 让 predecessor 的右指针指向 root，继续遍历左子树
            if (predecessor->right == NULL) {
                predecessor->right = root;
                root = root->left;
            }
            // 说明左子树已经访问完了，我们需要断开链接
            else {
                res[(*returnSize)++] = root->val;
                predecessor->right = NULL;
                root = root->right;
            }
        }
        // 如果没有左孩子，则直接访问右孩子
        else {
            res[(*returnSize)++] = root->val;
            root = root->right;
        }
    }
    return res;
}

时间复杂度：O(n)，其中 nn 为二叉搜索树的节点个数。Morris 遍历中每个节点会被访问两次，因此总 时间复杂度为O(2n)=O(n)。

空间复杂度：O(1)O(1)。