规范格式

This commit is contained in:
YDZ
2020-08-07 15:50:06 +08:00
parent 854a339abc
commit 4e11f4028a
1438 changed files with 907 additions and 924 deletions

View File

@ -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)
}

View File

@ -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")
}

View 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]
]
```
## 题目大意
按层序从上到下遍历一颗树。
## 解题思路
用一个队列即可实现。