题目描述:
给你一个整数数组 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]
提示:
2 <= nums.length <= 105
-30 <= nums[i] <= 30
保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内
进阶:你可以在 O(1) 的额外空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)
动态规划 c++代码:
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n=nums.size();
vector<int> ans;
int dp1[n],dp2[n];//dp1[i]表示第i个之前所有元素的乘积;dp2[i]表示之后所有元素乘积
dp1[0]=1,dp2[n-1]=1;
for(int i=1;i<n;i++){
dp1[i]=dp1[i-1]*nums[i-1];
}
for(int i=n-2;i>=0;i--){
dp2[i]=dp2[i+1]*nums[i+1];
}
for(int i=0;i<n;i++){
ans.push_back(dp1[i]*dp2[i]);
}
return ans;
}
};
进阶空间复杂度O(1) c++代码:
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n=nums.size();
vector<int> ans(n); //vector初始化时指定长度,就可以像int[]一样直接取下标而不用push_back了
ans[0]=1;
for(int i=1;i<n;i++){//先让ans[i]记录第i个前面的所有数乘积
ans[i]=ans[i-1]*nums[i-1];
}
int right=1;
for(int i=n-2;i>=0;i--){ //再乘上右边的所有数乘积
right=right*nums[i+1];
ans[i]=ans[i]*right;
}
return ans;
}
};
输出数组不占用空间复杂度,所以把所有运算都放在输出数组中,空间复杂度就为O(1)
总结:
输出不算空间复杂度,在输出数组中计算,空间复杂度为O(1)
vector在声明时,声明长度:vector v(n),就可以像int[]一样直接下标访问,不需要再用push_back初始化。