溫馨提示×

golang二叉樹層序遍歷怎么實現(xiàn)

小億
97
2023-11-22 03:01:09
欄目: 編程語言

要實現(xiàn)二叉樹的層序遍歷,可以使用隊列來輔助實現(xiàn)。

首先,我們需要定義二叉樹的結構:

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

然后,我們可以使用隊列來輔助層序遍歷:

func levelOrder(root *TreeNode) [][]int {
    if root == nil {
        return [][]int{}
    }

    result := [][]int{}
    queue := []*TreeNode{root}

    for len(queue) > 0 {
        levelSize := len(queue)
        level := []int{}

        for i := 0; i < levelSize; i++ {
            node := queue[0]
            queue = queue[1:]

            level = append(level, node.Val)
            if node.Left != nil {
                queue = append(queue, node.Left)
            }
            if node.Right != nil {
                queue = append(queue, node.Right)
            }
        }

        result = append(result, level)
    }

    return result
}

該函數(shù)首先檢查根節(jié)點是否為空,如果為空則直接返回空的結果。

然后,我們創(chuàng)建一個隊列,并將根節(jié)點放入隊列中。

我們進入一個循環(huán),直到隊列為空。在每次循環(huán)中,我們記錄當前隊列的大小,這樣我們就可以知道當前層級的節(jié)點數(shù)。

我們從隊列中取出當前層級的節(jié)點,并將其值加入到該層級的結果數(shù)組中。然后,我們將該節(jié)點的左右子節(jié)點加入到隊列中,以便下一層的遍歷。

最后,我們將該層級的結果數(shù)組加入到最終結果數(shù)組中。

最后,返回最終結果數(shù)組就是二叉樹的層序遍歷結果。

請注意,以上代碼僅提供了一種實現(xiàn)方式。在實際使用中,可能有不同的實現(xiàn)方式。

0