二叉树的中序遍历----递归、循环、以及空间复杂度O(1)解法

话不多说,前两种直接上代码:

递归方法:

void preOrder(TreeNode* root){
    if(root != NULL){
        preOrder(root->left);
        cout << root->val << " ";
        preOrder(root->right);
    }
}

循环方法:

void preOrder(TreeNode* root){
    if(root == NULL) return;
    stack<TreeNode*> S;
    TreeNode* node = root;
    while(node!=NULL || !S.empty()){
    	if(node != NULL)}{
    	     // cout << node->val << " ";  先序遍历:在push时访问结点值
 			 S.push(node);
        	  node = node->left;
    	}
    	else{
            cout << node->val << " ";   // 中序遍历:在弹出栈顶指针时访问
            node = S.top()->right;
            S.pop();
    	}  
    }
}

上述两种方法,时间复杂度为O(nlgn),空间复杂度为O(n)

优化空间复杂度为O(1)的方法:利用线索二叉树的概念

如果不能用栈空间,一个明显的问题是:如何返回到父节点。

线索二叉树利用叶子节点中的左右空指针指向某种顺序遍历下的前驱节点或后继节点就可以。

步骤如下:

1. 如果当前节点的左孩子为空,则输出当前节点并将其右孩子作为当前节点。

2. 如果当前节点的左孩子不为空,在当前节点的左子树中找到当前节点在中序遍历下的前驱节点。 a) 如果前驱节点的右孩子为空,将它的右孩子设置为当前节点。当前节点(Cur)更新为当前节点的左孩子
注意(不是改变节点内容,而是把游标Cur继续行进到左孩子)。 b) 如果前驱节点的右孩子为当前节点,将它的右孩子重新设为空(恢复树的形状)。输出当前节点。当前节点更新为当前节点的右孩子。

3. 重复以上1、2直到当前节点(Cur)为空。

画图示意:

图片和步骤来源:https://www.cnblogs.com/AnnieKim/archive/2013/06/15/MorrisTraversal.html

具体代码如下:

    void recoverTree(TreeNode *root)///这里是采用O(1)的算法
    {
        if(root==NULL)
            return;
        TreeNode* pre=NULL;
        TreeNode* cur=root;
        while(cur)
        {
            if(cur->left==NULL)
            {
                cout<<cur->val;
                cur=cur->right;
            }
            else
            {
                pre=cur->left;
                while(pre->right!=NULL&&pre->right!=cur)
                    pre=pre->right;
                if(pre->right==NULL)
                {
                    pre->right=cur;
                    cur=cur->left;
                }
                else
                {
                    pre->right=NULL;
                    cout<<cur->val;
                    cur=cur->right;
                }
            }
        }
    }

 

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
已知二叉树的前序遍历和后序遍历,可以推导出中序遍历。但需要注意的是,只有给定前序遍历和后序遍历是无法唯一确定一棵二叉树的,因此以下的解法也仅给出一种可能的中序遍历。 具体做法如下: 1. 根据前序遍历的第一个节点,确定根节点。 2. 在后序遍历中找到该根节点的位置,可以将后序遍历分为两个部分:左子树的后序遍历和右子树的后序遍历。 3. 在前序遍历中,根据左子树的节点个数,可以将前序遍历分为三个部分:根节点、左子树的前序遍历和右子树的前序遍历。 4. 对左子树和右子树分别递归进行上述步骤,直到遍历完所有节点。 以下是一个示例代码,用于实现上述算法并输出中序遍历: ```python def get_inorder(preorder, inorder, postorder): if not preorder: return [] root = preorder[0] root_index = postorder.index(root) left_size = root_index + 1 left_preorder = preorder[1:1+left_size] left_inorder = inorder[:left_size-1] left_postorder = postorder[:root_index+1] right_preorder = preorder[1+left_size:] right_inorder = inorder[left_size:] right_postorder = postorder[root_index+1:-1] left_inorder = get_inorder(left_preorder, left_inorder, left_postorder) right_inorder = get_inorder(right_preorder, right_inorder, right_postorder) return left_inorder + [root] + right_inorder preorder = [1, 2, 4, 5, 3, 6, 7] inorder = [4, 2, 5, 1, 6, 3, 7] postorder = [4, 5, 2, 6, 7, 3, 1] print(get_inorder(preorder, inorder, postorder)) # 输出 [4, 2, 5, 1, 6, 3, 7] ``` 在上述代码中,`get_inorder` 函数接收三个参数:前序遍历、中序遍历和后序遍历。如果前序遍历为空,则返回一个空列表。否则,首先根据前序遍历的第一个节点确定根节点,然后在后序遍历中找到该根节点的位置,并计算出左子树的节点个数。接着,根据左子树的节点个数,将前序遍历、中序遍历和后序遍历分为左子树和右子树三部分。对左子树和右子树分别递归进行上述步骤,并将结果合并,得到最终的中序遍历
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值