给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4] 输出: [24,12,8,6] 说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶: 你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路一:
这个题如果单纯的实现功能还是简单的,但是要求特别多。先暂且不管进阶要求,题目要求是不能使用除法,而且要在O(n)的时间复杂度。说明不能循环套循环。
思路是使用动态规划,新建两个数组L和R,分别存储第i个节点的左边所有数的乘积和右边所有数的乘积。然后最终结果,第i个数的除自身以外乘积就是L[i]*R[i]。
- 时间复杂度:O(n)
- 空间复杂度:O(n)
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
if(len==0) return nums;
int[] L = new int[len]; //左边数的乘积
int[] R = new int[len]; //右边数的乘积
L[0] = 1;
R[len-1] = 1;
for(int i=1;i<len;i++){
L[i] = L[i-1]*nums[i-1];
R[len-i-1] = R[len-i]*nums[len-i];
}
int[] res = new int[len];
for(int i=0;i<len;i++){
res[i] = L[i]*R[i];
}
return res;
}
}
思路二:
那么怎么才能不使用额外的空间呢,题目说最后输出的数组不算额外空间。那么我们就把它既当作最后输出的数组,又当作原来的L。那么原来的R怎么办呢,可以只用一个int来表示,用完一次就更新。具体步骤如下:
- 构造出左侧数乘积数组(即原来的L),取名res;
- 从res的末尾开始更新res为最终结果。用R来表示当前数的右边所有数的乘积。即
res[i] = res[i]*R;
R = R*nums[i];
- 最后返回res。
class Solution {
public int[] productExceptSelf(int[] nums) {
int len = nums.length;
if(len==0) return nums;
int[] res = new int[len];
res[0] = 1;
for(int i=1;i<len;i++){
res[i] = res[i-1]*nums[i-1];
}
int r = 1;
for(int i=len-1;i>=0;i--){
res[i] = res[i]*r;
r = r*nums[i];
}
return res;
}
}