题目
给你一个整数数组 nums
,返回 数组 answer
,其中 answer[i]
等于 nums
中除 nums[i]
之外其余各元素的乘积 。
题目数据保证数组 nums
之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。
请不要使用除法,且在 O(n)
时间复杂度内完成此题。
示例 1:
输入: nums = [1,2,3,4] 输出: [24,12,8,6]
示例 2:
输入: nums = [-1,1,0,-3,3] 输出: [0,0,9,0,0]
C++代码
#include <iostream>
#include <vector>
using namespace std;
/*
* 定义一个函数返回其余各元素的乘积
* 定义L、R两个数组分别存放当前元素左侧和右侧的各元素乘积
* 第一个元素的左侧没有元素,定义L[1]=1,求出L[i]
* 右侧同理,那么最后answer[i]等于L[i]和R[i]的乘积
*/
vector<int> productExceptSelf(vector<int>& nums) {
int n = nums.size();
vector<int> L(n), R(n), answer(n);
L[0] = 1;
for (int i = 1; i < n; i++) {
L[i] = nums[i - 1] * L[i - 1];
}
R[n - 1] = 1;
for (int i = n - 2; i >= 0; i--) {
R[i] = nums[i + 1] * R[i + 1];
}
for (int i = 0; i < n; i++) {
answer[i] = L[i] * R[i];
}
return answer;
}
int main() {
vector<int> nums = { 1,2,3,4 };
cout << "The nums: ";
for (int i = 0; i < nums.size(); i++) {
cout << nums[i] << " ";
}
cout << endl;
vector<int> answer = productExceptSelf(nums);
cout << "The product: ";
for (int i = 0; i < answer.size(); i++) {
cout << answer[i] << " ";
}
cout << endl;
return 0;
}
分析
定义一个函数 productExceptSelf() 返回其余各元素的乘积,定义 L、R 两个数组分别存放当前元素左侧和右侧的各元素乘积,第一个元素的左侧没有元素,定义 L[1] = 1,求出 L[i]。右侧同理,那么最后 answer[i] 等于 L[i] 和 R[i] 的乘积。