优化238

Author: diguage

docs/0238-product-of-array-except-self.adoc

@@ -35,6 +35,9 @@ https://leetcode.cn/problems/product-of-array-except-self/[LeetCode - 238. 除
 
 == 思路分析
 
+利用 xref:0000-03-prefix-sum.adoc[Prefix Sum 前缀和] 的思路，分别求前缀乘积和后缀乘积，然后再逐个求每个元素的前后元素乘积。
+
+利用结果数组存前缀乘积，可以将空间复杂度降低到常数级。
 
 [[src-0238]]
 [tabs]

logbook/202503.adoc

@@ -1767,7 +1767,7 @@ endif::[]
 |{counter:codes2503}
 |{leetcode_base_url}/product-of-array-except-self/[238. 除自身以外数组的乘积^]
 |{doc_base_url}/0238-product-of-array-except-self.adoc[题解]
-|✅ 前缀和。利用前缀和的思路，分别求前缀乘积和后缀乘积，然后再逐个求每个元素的前后元素乘积。
+|✅ 前缀和。利用前缀和的思路，分别求前缀乘积和后缀乘积，然后再逐个求每个元素的前后元素乘积。利用结果数组存前缀乘积，可以将空间复杂度降低到常数级。
 
 |===
 

src/main/java/com/diguage/algo/leetcode/_0238_ProductOfArrayExceptSelf_3.java

@@ -1,29 +1,25 @@
 package com.diguage.algo.leetcode;
 
-import java.util.Arrays;
-
 public class _0238_ProductOfArrayExceptSelf_3 {
   // tag::answer[]
+
   /**
    * @author D瓜哥 · https://www.diguage.com
    * @since 2025-10-22 19:31:52
    */
   public int[] productExceptSelf(int[] nums) {
-    int[] prefix = new int[nums.length];
-    Arrays.fill(prefix, 1);
-    int[] suffix = new int[nums.length];
-    Arrays.fill(suffix, 1);
-    for (int i = 0; i < nums.length - 1; i++) {
-      prefix[i] = (i - 1 >= 0 ? prefix[i - 1] : 1) * nums[i];
+    // 先使用 result 存储前缀乘积
+    int[] result = new int[nums.length];
+    result[0] = nums[0];
+    for (int i = 1; i < nums.length - 1; i++) {
+      result[i] = result[i - 1] * nums[i];
     }
+    int suffix = 1; // 利用变量 suffix 存放后缀乘积
     for (int i = nums.length - 1; i > 0; i--) {
-      suffix[i] = (i + 1 < nums.length ? suffix[i + 1] : 1) * nums[i];
-    }
-    int[] result = new int[nums.length];
-    for (int i = 0; i < nums.length; i++) {
-      result[i] = (i - 1 >= 0 ? prefix[i - 1] : 1)
-        * (i + 1 < nums.length ? suffix[i + 1] : 1);
+      result[i] = result[i - 1] * suffix;
+      suffix *= nums[i];
     }
+    result[0] = suffix;
     return result;
   }