二叉树中序遍历迭代实现

hust_joker

于 2021-05-27 09:26:44 发布

阅读量114

点赞数

CC 4.0 BY-SA版权

分类专栏： go algorithms 二叉树文章标签：算法 leetcode golang

本文链接：https://blog.youkuaiyun.com/hust_joker/article/details/117320422

go 同时被 3 个专栏收录

16 篇文章

订阅专栏

二叉树

15 篇文章

订阅专栏

algorithms

13 篇文章

订阅专栏

该博客主要介绍了如何使用栈实现二叉树的中序遍历。定义了一个`Stack`结构体，包含一个节点数组和一个索引，实现了`IsEmpty`、`Push`和`Pop`方法。`inorderTraversal`函数通过迭代方式完成中序遍历，返回一个整数数组表示遍历结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
type Stack struct {
    Parr [1000]*TreeNode
    Index int
}

func(s *Stack) IsEmpty() bool {
    if s.Index <= 0 {
        return true
    }else {
        return false
    }
}

func(s *Stack) Push(p *TreeNode) {
    s.Parr[s.Index] = p
    s.Index = s.Index + 1
}

func(s *Stack) Pop() (*TreeNode){
    if s.Index > 0 {
        s.Index = s.Index - 1
        return s.Parr[s.Index]
    }else {
        return nil
    }
}

func inorderTraversal(root *TreeNode) []int {
    rsp := make([]int, 0)
    if nil == root {
        return rsp
    }

    stack := Stack{}
  
    for ; !stack.IsEmpty() || root != nil; {
        for ; root != nil; {
            stack.Push(root)
            root = root.Left
        }

        if !stack.IsEmpty() {
            root = stack.Pop()
            rsp = append(rsp, root.Val)
            root = root.Right
        }

    }

    return rsp
}