题目描述:
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
示例:
输入: [1,2,3,4]
输出: [24,12,8,6]
提示:题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
思路:
一开始想用除法, O(N)不是也分分钟,但是此处不能使用,所以一开始就想到了矩阵的上三角和下三角,当前位置结果为左边乘数*右边成熟。
第一遍的解法使用了矩阵,使用i,j作为指针,一遍从左边遍历到右,一遍从右遍历到左,就可以填充完所有值,然后result[i] = dp[0][i-1]*dp[i+1][n]就可以获得结果,但是此处有提示 使用常数空间复杂度,查看题解后发现了更右解,直接乘数即可,不用再多矩阵的空间复杂度。
代码:
public static int[] productExceptSelf(int[] nums) {
int length = nums.length;
int i = 1;
int[] result = new int[length];
for (int k = 0; k < length; k++) {
result[k] = i;
i *= nums[k];
}
i = 1;
for (int k = length - 1; k >= 0; k--) {
result[k] *= i;
i *= nums[k];
}
return result;
}