给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
空间换时间。
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> output;
vector<int> arr1(nums.size(), 1);
vector<int> arr2(nums.size(), 1);
for(int i = 1; i < nums.size(); i++)
arr1[i] = arr1[i - 1] * nums[i - 1];
for(int i = nums.size() - 2; i >= 0; i--)
arr2[i] = arr2[i + 1] * nums[i + 1];
for(int j = 0; j < nums.size(); j++)
output.push_back(arr1[j] * arr2[j]);
return output;
}
时间复杂度:o(n)
空间复杂度:o(n)
优化版本:这里输出数组不计。
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> output(nums.size(), 1);
int tmpRight = 1;
for(int i = 1; i < nums.size(); i++)
output[i] = output[i - 1] * nums[i - 1];
for(int j = nums.size() - 1; j >= 0; j--) {
output[j] *= tmpRight;
tmpRight *= nums[j];
}
return output;
}
空间复杂度:O(1)