-
暴力法
最简单的方法时暴力法,两层遍历就可以搞定,但不符合题目的O(n)时间复杂度。 -
记录前后遍历结果
利用空间换时间的思想,分别建立两个数组来保存数组从前向后遍历和从后向前遍历时的各个元素的乘积,最后在利用一次遍历求出某个元素前、后乘积的乘积即为所需要的结果。
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
int[] leftsweep = new int[len];
int[] rightsweep = new int[len];
int[] output = new int[len];
leftsweep[0] = nums[0];
for(int i=1; i<len; i++){
leftsweep[i] = nums[i]*leftsweep[i-1];
}
rightsweep[len-1] = nums[len-1];
for(int i=len-2; i>=0; i--){
rightsweep[i] = nums[i]*rightsweep[i+1];
}
for(int i=0; i<len;i++){
if(i==0){
output[i] = rightsweep[i+1];
}else if(i==len-1){
output[i] = leftsweep[i-1];
}else{
output[i] = leftsweep[i-1]*rightsweep[i+1];
}
}
return output;
}
}
- O(1)空间复杂度
实际上我们可以直接利用结果数组来存储两次前、后遍历的结果:
class Solution {
public int[] productExceptSelf(int[] nums) {
int[] res = new int[nums.length];
int k = 1;
for(int i = 0; i < res.length; i++){
res[i] = k;
k = k * nums[i]; // 此时数组存储的是除去当前元素左边的元素乘积
}
k = 1;
for(int i = res.length - 1; i >= 0; i--){
res[i] *= k; // k为该数右边的乘积。
k *= nums[i]; // 此时数组等于左边的 * 该数右边的。
}
return res;
}
}