mirror of
https://github.com/halfrost/LeetCode-Go.git
synced 2025-07-05 00:25:22 +08:00
规范格式
This commit is contained in:
@ -0,0 +1,70 @@
|
||||
package leetcode
|
||||
|
||||
import (
|
||||
"github.com/halfrost/LeetCode-Go/structures"
|
||||
)
|
||||
|
||||
// TreeNode define
|
||||
type TreeNode = structures.TreeNode
|
||||
|
||||
/**
|
||||
* Definition for a binary tree node.
|
||||
* type TreeNode struct {
|
||||
* Val int
|
||||
* Left *TreeNode
|
||||
* Right *TreeNode
|
||||
* }
|
||||
*/
|
||||
|
||||
// 解法一 BFS
|
||||
func levelOrder(root *TreeNode) [][]int {
|
||||
if root == nil {
|
||||
return [][]int{}
|
||||
}
|
||||
queue := []*TreeNode{}
|
||||
queue = append(queue, root)
|
||||
curNum, nextLevelNum, res, tmp := 1, 0, [][]int{}, []int{}
|
||||
for len(queue) != 0 {
|
||||
if curNum > 0 {
|
||||
node := queue[0]
|
||||
if node.Left != nil {
|
||||
queue = append(queue, node.Left)
|
||||
nextLevelNum++
|
||||
}
|
||||
if node.Right != nil {
|
||||
queue = append(queue, node.Right)
|
||||
nextLevelNum++
|
||||
}
|
||||
curNum--
|
||||
tmp = append(tmp, node.Val)
|
||||
queue = queue[1:]
|
||||
}
|
||||
if curNum == 0 {
|
||||
res = append(res, tmp)
|
||||
curNum = nextLevelNum
|
||||
nextLevelNum = 0
|
||||
tmp = []int{}
|
||||
}
|
||||
}
|
||||
return res
|
||||
}
|
||||
|
||||
// 解法二 DFS
|
||||
func levelOrder1(root *TreeNode) [][]int {
|
||||
levels := [][]int{}
|
||||
dfsLevel(root, -1, &levels)
|
||||
return levels
|
||||
}
|
||||
|
||||
func dfsLevel(node *TreeNode, level int, res *[][]int) {
|
||||
if node == nil {
|
||||
return
|
||||
}
|
||||
currLevel := level + 1
|
||||
for len(*res) <= currLevel {
|
||||
*res = append(*res, []int{})
|
||||
}
|
||||
(*res)[currLevel] = append((*res)[currLevel], node.Val)
|
||||
dfsLevel(node.Left, currLevel, res)
|
||||
dfsLevel(node.Right, currLevel, res)
|
||||
}
|
@ -0,0 +1,56 @@
|
||||
package leetcode
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"github.com/halfrost/LeetCode-Go/structures"
|
||||
)
|
||||
|
||||
type question102 struct {
|
||||
para102
|
||||
ans102
|
||||
}
|
||||
|
||||
// para 是参数
|
||||
// one 代表第一个参数
|
||||
type para102 struct {
|
||||
one []int
|
||||
}
|
||||
|
||||
// ans 是答案
|
||||
// one 代表第一个答案
|
||||
type ans102 struct {
|
||||
one [][]int
|
||||
}
|
||||
|
||||
func Test_Problem102(t *testing.T) {
|
||||
|
||||
qs := []question102{
|
||||
|
||||
question102{
|
||||
para102{[]int{}},
|
||||
ans102{[][]int{}},
|
||||
},
|
||||
|
||||
question102{
|
||||
para102{[]int{1}},
|
||||
ans102{[][]int{[]int{1}}},
|
||||
},
|
||||
|
||||
question102{
|
||||
para102{[]int{3, 9, 20, structures.NULL, structures.NULL, 15, 7}},
|
||||
ans102{[][]int{[]int{3}, []int{9, 20}, []int{15, 7}}},
|
||||
},
|
||||
}
|
||||
|
||||
fmt.Printf("------------------------Leetcode Problem 102------------------------\n")
|
||||
|
||||
for _, q := range qs {
|
||||
_, p := q.ans102, q.para102
|
||||
fmt.Printf("【input】:%v ", p)
|
||||
root := structures.Ints2TreeNode(p.one)
|
||||
fmt.Printf("【output】:%v \n", levelOrder(root))
|
||||
}
|
||||
fmt.Printf("\n\n\n")
|
||||
}
|
39
leetcode/0102.Binary-Tree-Level-Order-Traversal/README.md
Normal file
39
leetcode/0102.Binary-Tree-Level-Order-Traversal/README.md
Normal file
@ -0,0 +1,39 @@
|
||||
# [102. Binary Tree Level Order Traversal](https://leetcode.com/problems/binary-tree-level-order-traversal/)
|
||||
|
||||
## 题目
|
||||
|
||||
|
||||
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
|
||||
|
||||
For example:
|
||||
Given binary tree [3,9,20,null,null,15,7],
|
||||
|
||||
```c
|
||||
3
|
||||
/ \
|
||||
9 20
|
||||
/ \
|
||||
15 7
|
||||
```
|
||||
|
||||
return its level order traversal as:
|
||||
|
||||
```c
|
||||
[
|
||||
[3],
|
||||
[9,20],
|
||||
[15,7]
|
||||
]
|
||||
```
|
||||
|
||||
|
||||
## 题目大意
|
||||
|
||||
按层序从上到下遍历一颗树。
|
||||
|
||||
## 解题思路
|
||||
|
||||
用一个队列即可实现。
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user