栈模拟递归 遍历二叉树的正确写法


对于二叉树的生成,遍历,应该是树这个数据结构需要的基本功,只有真的理解了树的生成,遍历,递归的过程,才能更好的去做二叉树的翻转等题目。

二叉树的生成

1.通过层次遍历来顺序插入

2.利用左右节点进行插入,自由度最高,可以形成任意形状的树

# 节点结构和树结构:
class Node():
    def __init__(self, val: int):
        self.val = val
        self.lchild = None
        self.rchild = None


class Tree():
    def __init__(self, root: Node = None):
        self.root = root
        self.q = []  # 插入时顺序插入的话,需要在树的属性中保存一个当前节点状态的队列,\
        # 以便获取当前插入节点的父节点

    def insert(self, node: Node):
        if not self.root:
            self.root = node
            self.q.append(self.root)
        else:
            parNode = self.q[0]
            if not parNode.lchild:
                parNode.lchild = node
                self.q.append(node)
            elif not parNode.rchild:
                parNode.rchild = node
                self.q.append(node)
                self.q.pop(0)
  

这个写法的好处是,完全支持两种初始化的方式:

在这里插入图片描述

生成如图所示的节点:

# method1 层次顺序插入
tree = Tree()
for i in range(6):
	node = Node(i)
    tree.insert(node)
  
# method2 直接构建树,支持随意插入
root = Node(0)
_tree 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
二叉树是一种重要的数据结构,在许多算法中都有广泛的应用。二叉树遍历是指按照某种规则依次访问二叉树中的所有节点。常用的遍历方法有前序遍历、中序遍历和后序遍历。在本文中,我们将介绍递归遍历二叉树和非递归遍历二叉树的设计思想。 一、递归遍历二叉树的设计思想 递归遍历二叉树的基本思想是将二叉树的节点拆分成两个部分:根节点和子树。对于每个节点,我们先访问它的根节点,然后分别对其左右子树进行递归遍历。具体实现过程如下: 1. 前序遍历 前序遍历的顺序是先访问根节点,然后访问左子树,最后访问右子树。代码实现如下: ``` void preOrder(TreeNode* root) { if(root == NULL) return; cout << root->val << " "; // 访问根节点 preOrder(root->left); // 递归遍历左子树 preOrder(root->right); // 递归遍历右子树 } ``` 2. 中序遍历 中序遍历的顺序是先访问左子树,然后访问根节点,最后访问右子树。代码实现如下: ``` void inOrder(TreeNode* root) { if(root == NULL) return; inOrder(root->left); // 递归遍历左子树 cout << root->val << " "; // 访问根节点 inOrder(root->right); // 递归遍历右子树 } ``` 3. 后序遍历 后序遍历的顺序是先访问左子树,然后访问右子树,最后访问根节点。代码实现如下: ``` void postOrder(TreeNode* root) { if(root == NULL) return; postOrder(root->left); // 递归遍历左子树 postOrder(root->right); // 递归遍历右子树 cout << root->val << " "; // 访问根节点 } ``` 递归遍历二叉树的优点是代码简洁明了,易于理解。然而,递归遍历二叉树的缺点是可能会导致溢出问题。当二叉树的高度很大时,递归遍历可能会占用大量的系统空间,导致程序崩溃。因此,我们需要使用非递归遍历二叉树的方法来避免这个问题。 二、非递归遍历二叉树的设计思想 非递归遍历二叉树的基本思想是使用模拟递归遍历的过程。在递归遍历二叉树时,系统会自动为每个递归函数分配一段空间,用于保存函数的局部变量和返回地址。而在非递归遍历二叉树时,我们需要手动维护一个来存储访问过的节点。具体实现过程如下: 1. 前序遍历 前序遍历的非递归实现可以使用一个来保存节点。首先将根节点入,然后进入循环,将顶节点弹出并输出,然后将右子树入,最后将左子树入。这样就可以按照前序遍历的顺序遍历二叉树。代码实现如下: ``` void preOrder(TreeNode* root) { stack<TreeNode*> s; if(root != NULL) s.push(root); // 根节点入 while(!s.empty()) { TreeNode* node = s.top(); // 取出顶节点 s.pop(); cout << node->val << " "; // 输出节点值 if(node->right != NULL) s.push(node->right); // 右子树入 if(node->left != NULL) s.push(node->left); // 左子树入 } } ``` 2. 中序遍历 中序遍历的非递归实现也可以使用一个来保存节点。首先将根节点入,然后进入循环,如果不为空或者当前节点不为空,就执行以下步骤:将当前节点的左子树依次入,取出顶节点并输出,将当前节点指向其右子树。这样就可以按照中序遍历的顺序遍历二叉树。代码实现如下: ``` void inOrder(TreeNode* root) { stack<TreeNode*> s; TreeNode* node = root; while(!s.empty() || node != NULL) { if(node != NULL) { // 当前节点不为空,将左子树入 s.push(node); node = node->left; } else { // 当前节点为空,取出顶节点并输出 node = s.top(); s.pop(); cout << node->val << " "; node = node->right; // 将当前节点指向其右子树 } } } ``` 3. 后序遍历 后序遍历的非递归实现需要使用两个来保存节点。首先将根节点入1,然后进入循环,从1取出一个节点,将该节点的左右子树分别入1,然后将该节点入2。重复以上步骤,直到1为空。然后依次从2中取出节点并输出。这样就可以按照后序遍历的顺序遍历二叉树。代码实现如下: ``` void postOrder(TreeNode* root) { stack<TreeNode*> s1, s2; if(root != NULL) s1.push(root); // 根节点入1 while(!s1.empty()) { TreeNode* node = s1.top(); // 取出顶节点 s1.pop(); s2.push(node); // 将节点入2 if(node->left != NULL) s1.push(node->left); // 左子树入1 if(node->right != NULL) s1.push(node->right); // 右子树入1 } while(!s2.empty()) { TreeNode* node = s2.top(); // 取出顶节点并输出 s2.pop(); cout << node->val << " "; } } ``` 非递归遍历二叉树的优点是可以避免溢出问题,而且效率比递归遍历要高。因此,在实际的程序设计中,我们更加倾向于使用非递归遍历二叉树的方法。 总结 本文介绍了递归遍历二叉树和非递归遍历二叉树的设计思想。递归遍历二叉树的基本思想是将二叉树的节点拆分成根节点和子树,然后对每个节点进行递归遍历。非递归遍历二叉树的基本思想是使用模拟递归遍历的过程,依次访问每个节点。在实际的程序设计中,我们可以根据具体的需求选择适合的遍历方式。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值