-
Notifications
You must be signed in to change notification settings - Fork 3
/
solution.go
47 lines (43 loc) · 819 Bytes
/
solution.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package main
import "fmt"
// TreeNode is a binary tree node
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
func maxDepth(root *TreeNode) int {
if root == nil {
return 0
}
answer := 1
var getMaxDepth func(root *TreeNode, depth int)
getMaxDepth = func(root *TreeNode, depth int) {
if root == nil {
return
}
if root.Left == nil && root.Right == nil {
if depth > answer {
answer = depth
}
}
getMaxDepth(root.Left, depth+1)
getMaxDepth(root.Right, depth+1)
}
getMaxDepth(root, answer)
return answer
}
func main() {
tree := &TreeNode{1,
&TreeNode{2,
&TreeNode{4,
&TreeNode{7, nil, nil},
nil},
&TreeNode{5, nil, nil}},
&TreeNode{3,
&TreeNode{6,
&TreeNode{8, nil, nil},
&TreeNode{9, nil, nil}},
nil}}
fmt.Println(maxDepth(tree))
}