This commit is contained in:
parent
84963a62a3
commit
e36dc6a584
|
|
@ -0,0 +1,25 @@
|
||||||
|
```go
|
||||||
|
func countNodes(root *TreeNode) int {
|
||||||
|
if root == nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
q := list.New()
|
||||||
|
q.PushBack(root)
|
||||||
|
res := 0
|
||||||
|
for q.Len() > 0 {
|
||||||
|
n := q.Len()
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
node := q.Remove(q.Front()).(*TreeNode)
|
||||||
|
if node.Left != nil {
|
||||||
|
q.PushBack(node.Left)
|
||||||
|
}
|
||||||
|
if node.Right != nil {
|
||||||
|
q.PushBack(node.Right)
|
||||||
|
}
|
||||||
|
res++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
|
@ -0,0 +1,22 @@
|
||||||
|
```go
|
||||||
|
func maxDepth(root *Node) int {
|
||||||
|
if root == nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
q := list.New()
|
||||||
|
q.PushBack(root)
|
||||||
|
depth := 0
|
||||||
|
for q.Len() > 0 {
|
||||||
|
n := q.Len()
|
||||||
|
for i := 0; i < n; i++ {
|
||||||
|
node := q.Remove(q.Front()).(*Node)
|
||||||
|
for j := range node.Children {
|
||||||
|
q.PushBack(node.Children[j])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
depth++
|
||||||
|
}
|
||||||
|
return depth
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
Loading…
Reference in New Issue