✨543.二叉树的直径

Author: techoc

README.md

@@ -60,6 +60,7 @@
 - [387.字符串中的第一个唯一字符](leetcode/easy/387.go)
 - [415.字符串相加](leetcode/easy/t415/solution.go)
 - [521.最长特殊序列 Ⅰ](leetcode/easy/521.go)
+- [543.二叉树的直径](leetcode/easy/t543/solution.go)
 - [566.重塑矩阵](leetcode/easy/556.go)
 - [590.N 叉树的后序遍历](leetcode/easy/590.go)
 - [653.两数之和 IV-输入 BST](leetcode/easy/653.go)

leetcode/easy/t543/solution.go

@@ -0,0 +1,25 @@
+package t543
+
+import (
+	. "github.com/techoc/leetcodego/leetcode/utils"
+)
+
+// 543. 二叉树的直径
+// https://leetcode.cn/problems/diameter-of-binary-tree/
+func diameterOfBinaryTree(root *TreeNode) int {
+	var res = 0                   // 初始化结果变量为0
+	var depth func(*TreeNode) int // 声明一个内部函数，用于计算节点的深度
+
+	// 定义depth函数，用于递归计算二叉树的深度并更新结果
+	depth = func(node *TreeNode) int {
+		if node == nil {
+			return 0 // 如果节点为空，深度为0
+		}
+		l := depth(node.Left)  // 计算左子树的深度
+		r := depth(node.Right) // 计算右子树的深度
+		res = max(res, l+r)    // 更新结果为左右子树深度之和的最大值
+		return max(l, r) + 1   // 返回节点的深度，即左右子树最大深度加一
+	}
+	depth(root) // 从根节点开始计算深度
+	return res  // 返回结果
+}

leetcode/easy/t543/solution_test.go

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