LeetCode笔记:238. Product of Array
问题:
Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].
Solve it without division and in O(n).
For example, given [1,2,3,4], return [24,12,8,6].
Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)
大意:
给出一个有n(n>1)个整数的数组nums,返回一个output数组,output[i]等于除了nums[i]外其余所有元素的乘积。
不使用除法且在O(n)时间内完成。
比如,给出 [1,2,3,4],返回 [24,12,8,6]。
进阶:
你能使用固定的空间复杂度吗?(注意:output数组不算做额外的空间。)
思路:
如果用除法就简单了,直接全部乘起来,然后每个位置对应除以nums[i]的元素就可以了。
不用除法的话,我们要用两次遍历,先正着遍历一遍,在结果数组上每个元素都算到累乘至nums数组中对应位置的前面所有的元素,比如第三个元素的值为nums中前连个元素的乘积。
第二次遍历,反着遍历,用一个变量记录从后到前的累乘,同时结果数组中乘以这个变量。
这样对每一个位置来说,其刚好在第一次遍历中取得了它前面所有元素的乘积,第二次遍历中又乘以了它后面所有元素的乘积,唯独不算它自己在内。
代码(Java):
public class Solution {
public int[] productExceptSelf(int[] nums) {
int[] result = new int[nums.length];
for (int i = 0; i < result.length; i++) result[i] = 1;
for (int i = 1; i < nums.length; i++) {// 先正着来一遍,只乘到前一个元素
result[i] = result[i-1] * nums[i-1];
}
int behind = 1;
for (int i = nums.length-1; i >= 0; i--) {// 再倒着来一遍,乘以后面的数
result[i] = result[i] * behind;
behind = behind * nums[i];
}
return result;
}
}
合集:https://github.com/Cloudox/LeetCode-Record