力扣94：二叉树的中序遍历

最新推荐文章于 2024-12-24 23:56:39 发布

原创最新推荐文章于 2024-12-24 23:56:39 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

基础专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种使用Go语言实现的二叉树中序遍历算法，通过定义二叉树节点结构并采用递归的方式进行遍历，最终输出二叉树的所有节点值。

这里用了匿名函数没用全局变量以便同时可以跑多个用例

package main

import "fmt"

type TreeNode struct {
	Val   int
	Left  *TreeNode
	Right *TreeNode
}

func inorderTraversal(root *TreeNode) (result []int) {
	var in func(node *TreeNode)
	in = func(node *TreeNode) {
		if node == nil {
			return
		}
		in(node.Left)
		result = append(result, node.Val)
		in(node.Right)
	}
	in(root)
	return
}

func main() {
	root := &TreeNode{Val: 1,
		Left: &TreeNode{Val: 2,
			Left:  &TreeNode{Val: 4},
			Right: &TreeNode{Val: 5}},
		Right: &TreeNode{Val: 3,
			Left:  &TreeNode{Val: 6},
			Right: &TreeNode{Val: 7}}}
	fmt.Println(inorderTraversal(root))

	node := &TreeNode{Val: 1,
		Left: &TreeNode{Val: 2,
			Left:  &TreeNode{Val: 4},
			Right: &TreeNode{Val: 5}},
		Right: &TreeNode{Val: 3,
			Left:  &TreeNode{Val: 6},
			Right: &TreeNode{Val: 7}}}
	fmt.Println(inorderTraversal(node))
}