Given an array of n integers where n > 1, return an array such that is equal to the product of all the elements of except . Example: Note: Please solve it without division and in O(n). Follow up: Could you solve it with constant space complexity? (The output array does not count as extra space for the purpose of space complexity analysis.) | 给定长度为 n 的整数数组 ,其中 n > 1,返回输出数组 ,其中 等于 中除 之外其余各元素的乘积。 示例: 说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。 进阶: 你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。) |
思路:题目说 不能用除法,而且必须O(n)。那就用两个数组,f[i]保存第i个元素之前所有元素 乘积,b[i]表示第i个 元素之后所有元素乘积,那第i个位置的除第i个位置值以外其余元素乘积为f[i]*b[i]; 比如[1,2,3,4],则f数组为[1,1,2,6],b数组元素为[24,12,8,6],刚好第i个 位置的值 就是f[i]*b[i];
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
if(nums.empty()) return {};
int n=nums.size();
vector<int> f(n,1),b(n,1),res(n,1);
for(int i=1;i<n;++i) f[i]=f[i-1]*nums[i-1];
for(int i=n-1;i>0;--i) b[i-1]=b[i]*nums[i];
for(int i=0;i<n;++i) res[i]=f[i]*b[i];
return res;
}
};