diff --git a/problems/0337.打家劫舍III.md b/problems/0337.打家劫舍III.md
index 12e31aba..a3130df7 100644
--- a/problems/0337.打家劫舍III.md
+++ b/problems/0337.打家劫舍III.md
@@ -477,14 +477,7 @@ func max(x, y int) int {
 ```go
 func rob(root *TreeNode) int {
 	res := robTree(root)
-	return max(res[0], res[1])
-}
-
-func max(a, b int) int {
-	if a > b {
-		return a
-	}
-	return b
+	return slices.Max(res)
 }
 
 func robTree(cur *TreeNode) []int {
@@ -498,7 +491,7 @@ func robTree(cur *TreeNode) []int {
     // 考虑去偷当前的屋子
 	robCur := cur.Val + left[0] + right[0]
     // 考虑不去偷当前的屋子
-	notRobCur := max(left[0], left[1]) + max(right[0], right[1])
+	notRobCur := slices.Max(left) + slices.Max(right)
 
     // 注意顺序：0:不偷，1:去偷
 	return []int{notRobCur, robCur}