✨104.二叉树的最大深度

Author: techoc

README.md

@@ -23,7 +23,7 @@
 - [94.二叉树的中序遍历](leetcode/easy/94.go)
 - [100.相同的树](leetcode/easy/100.go)
 - [101.对称二叉树](leetcode/easy/101.go)
-- [104.二叉树的最大深度](leetcode/easy/104.go)
+- [104.二叉树的最大深度](leetcode/easy/t104/solution.go)
 - [108.将有序数组转换为二叉搜索树](leetcode/easy/108.go)
 - [110.平衡二叉树](leetcode/easy/110.go)
 - [111.二叉树的最小深度](leetcode/easy/111.go)

leetcode/easy/104.go

@@ -0,0 +1,44 @@
+package t104
+
+import "github.com/techoc/leetcodego/leetcode/utils"
+
+/**
+ * Definition for a binary tree node.
+ * type TreeNode struct {
+ *     Val int
+ *     Left *TreeNode
+ *     Right *TreeNode
+ * }
+ */
+
+// 记录最大深度
+var res int
+
+// 记录遍历到的节点的深度
+var depth int
+
+// 104. 二叉树的最大深度
+// https://leetcode.cn/problems/maximum-depth-of-binary-tree/
+func maxDepth(root *utils.TreeNode) int {
+	// 初始化最大深度和节点深度 避免全局变量污染
+	res, depth = 0, 0
+	traverse(root)
+	return res
+}
+
+// 遍历二叉树
+func traverse(root *utils.TreeNode) {
+	if root == nil {
+		return
+	}
+	// 前序位置 节点深度加一
+	depth++
+	// 节点为叶子节点时，更新最大深度
+	if root.Left == nil && root.Right == nil {
+		res = max(res, depth)
+	}
+	traverse(root.Left)
+	traverse(root.Right)
+	// 后序位置 节点深度减一
+	depth--
+}

leetcode/easy/t104/solution.go

@@ -0,0 +1,32 @@
+package t104
+
+import (
+	"github.com/techoc/leetcodego/leetcode/utils"
+	"testing"
+)
+
+func TestMaxDepth(t *testing.T) {
+	tests := []struct {
+		name string
+		root *utils.TreeNode
+		want int
+	}{
+		{
+			name: "Case 1",
+			root: utils.ListToTree([]any{3, 9, 20, nil, nil, 15, 7}),
+			want: 3,
+		},
+		{
+			name: "Case 2",
+			root: utils.ListToTree([]any{1, nil, 2}),
+			want: 2,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			if got := maxDepth(tt.root); got != tt.want {
+				t.Errorf("maxDepth() = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}