题目描述
计算数组中除自身之外的其他元素的乘积。
示例:
Input: [1,2,3,4]
Output: [24,12,8,6]
要求:
(1)不使用除法
(2)时间复杂度为O(n)
(3)除最后返回的数组外,空间复杂度为O(1)
解题思路
不符合题意的两个方法
- 使用除法。首先计算出所有元素的总乘积,然后对于每个位置,都除以这个位置的元素,得到的就是除自身外其它元素的乘积。但是这个方法中要特别注意
0
的存在! - 使用左右乘积列表。
所谓除自身之外的乘积,其实就是该元素的左边所有元素相乘、右边所有元素相乘,得到的两个数再相乘的结果,那么自然而然就想到,提前遍历一遍数组,把各个位置左边元素相乘的结果和右边元素相乘的结果都分别记下来,最后对应相乘就可以。
注意这个方法的时间空间复杂度都为O(n),代码如下:
class Solution {
public int[] productExceptSelf(int[] nums) {
if (nums == null || nums.length == 0)
return nums;
int n = nums.length;
int[] left = new int[n];
int[] right = new int[n];
int tmp_left = 1, tmp_right = 1;
for (int i = 0; i < n; ++i) {
left[i] = tmp_left;
right[n - i - 1] = tmp_right;
tmp_left *= nums[i];
tmp_right *= nums[n - i - 1];
}
for (int i = 0; i < n; ++i) {
nums[i] = left[i] * right[i];
}
return nums;
}
}
改进(最后符合题意的方法)
本质思想和上面第二个方法还是一致的,左边元素乘积 * 右边元素乘积,不过不像上面一样用两个数组来保存,而是只用一个最终数组(题目中说了最后的结果数组不算在空间复杂度内)
新建一个数组,保存左边元素乘积,然后再次遍历nums
将右边元素乘积动态地乘进去,得到最终结果。
class Solution {
public int[] productExceptSelf(int[] nums) {
if (nums == null || nums.length == 0)
return nums;
int n = nums.length;
int product = 1; // 先保存左边元素乘积
int[] res = new int[n];
for (int i = 0; i < n; ++i) {
res[i] = product;
product *= nums[i];
}
product = 1; // 再保存右边元素乘积
for (int i = n - 1; i >= 0; --i) {
res[i] *= product;
product *= nums[i];
}
return res;
}
}