原题
给你一个长度为 n 的整数数组 nums
,其中 n > 1,返回输出数组 output
,其中 output[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积。
示例:
输入: [1,2,3,4] 输出: [24,12,8,6]
提示: 题目数据保证数组之中任意元素的全部前缀元素和后缀(甚至是整个数组)的乘积都在 32 位整数范围内。
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
解法
思想
顺序相反的两次遍历
代码
func productExceptSelf(nums []int) []int {
lsum := make([]int, len(nums))
rsum := make([]int, len(nums))
ans := make([]int, len(nums))
for index := 0; index < len(nums); index++{
if index == 0{
lsum[index] = 1
} else {
lsum[index] = lsum[index-1] * nums[index-1]
}
}
for index := len(nums) - 1; index >= 0; index--{
if index == len(nums) - 1{
rsum[index] = 1
} else {
rsum[index] = rsum[index+1] * nums[index+1]
}
}
for index := 0; index < len(nums); index++{
ans[index] = lsum[index] * rsum[index]
}
return ans
}
优化后:(减少空间复杂度,用ans代替lsum,用临时变量代替rsum)
func productExceptSelf(nums []int) []int {
length := len(nums)
answer := make([]int, length)
// answer[i] 表示索引 i 左侧所有元素的乘积
// 因为索引为 '0' 的元素左侧没有元素, 所以 answer[0] = 1
answer[0] = 1
for i := 1; i < length; i++ {
answer[i] = nums[i-1] * answer[i-1]
}
// R 为右侧所有元素的乘积
// 刚开始右边没有元素,所以 R = 1
R := 1
for i := length - 1; i >= 0; i-- {
// 对于索引 i,左边的乘积为 answer[i],右边的乘积为 R
answer[i] = answer[i] * R
// R 需要包含右边所有的乘积,所以计算下一个结果时需要将当前值乘到 R 上
R *= nums[i]
}
return answer
}
java:
class Solution {
public int[] productExceptSelf(int[] nums) {
int[] ans = new int[nums.length];
ans[0] = 1;
for(int i = 1;i<nums.length;i++){
ans[i] = ans[i-1]*nums[i-1];
}
int prev = nums[nums.length-1];
for(int i = nums.length-2;i>=0;i--){
ans[i] = prev*ans[i];
prev = prev*nums[i];
}
return ans;
}
}
原创文章,作者:彭晨涛,如若转载,请注明出处:https://www.codetool.top/article/leetcode238-%e9%99%a4%e8%87%aa%e8%ba%ab%e4%bb%a5%e5%a4%96%e6%95%b0%e7%bb%84%e7%9a%84%e4%b9%98%e7%a7%af/