给定长度为 n 的整数数组 nums
,其中 n > 1,返回输出数组 output
,其中 output[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积。
示例:
输入:[1,2,3,4]
输出:[24,12,8,6]
说明: 请不要使用除法,且在 O(n) 时间复杂度内完成此题。
进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
C
/**
* Return an array of size *returnSize.
* Note: The returned array must be malloced, assume caller calls free().
*/
int* productExceptSelf(int* nums, int numsSize, int* returnSize)
{
int n=numsSize;
int* left=(int*)malloc(sizeof(int)*n);
int* right=(int*)malloc(sizeof(int)*n);
int k=1;
for(int i=0;i<n;i++)
{
left[i]=k;
k*=nums[i];
}
k=1;
for(int i=n-1;i>=0;i--)
{
right[i]=k;
k*=nums[i];
}
for(int i=0;i<n;i++)
{
left[i]*=right[i];
}
*returnSize=n;
return left;
}
C++
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums)
{
int n=nums.size();
vector<int> left(n,1);
vector<int> right(n,1);
int k=1;
for(int i=0;i<n;i++)
{
left[i]=k;
k*=nums[i];
}
k=1;
for(int i=n-1;i>=0;i--)
{
right[i]=k;
k*=nums[i];
}
for(int i=0;i<n;i++)
{
left[i]*=right[i];
}
return left;
}
};
python
class Solution:
def productExceptSelf(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
n=len(nums)
left=[1 for i in range(n)]
right=[1 for i in range(n)]
k=1
for i in range(n):
left[i]=k
k*=nums[i]
k=1
for i in range(n-1,-1,-1):
right[i]=k
k*=nums[i]
for i in range(0,n):
left[i]*=right[i]
return left