力扣HOT100之普通数组:238. 除自身以外数组的乘积
这道题不能使用除法,我第一时间想到用前缀表和后缀表来解决,假设数组nums
的长度为n
,我们直接定义前缀表pre
和后缀表suf
,其中pre[i] = pre[i - 1] * nums[i - 1] = nums[0] * nums[1] * ... * nums[i - 1]
,而suf[j] = suf[j + 1] * nums[j + 1] = nums[n - 1] * nums[n - 2] * ... * nums[j + 1]
,所以返回的向量result
中,result[i] = pre[i] * suf[i]
,其中特别要注意的是,前缀表pre
的长度为n
,pre[0] = 1
,而后缀表的长度为n
,suf[n - 1] = 1
,一共需要三次线性遍历,时间复杂度为O(n)。
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> result;
vector<int> pre(nums.size(), 1); //前缀乘积
vector<int> suf(nums.size(), 1); //后缀乘积
for(int i = 1; i < pre.size(); ++i) //构造前缀乘积表
pre[i] = pre[i - 1] * nums[i - 1];
for(int i = suf.size() - 2; i >= 0; --i) //构造后缀乘积表
suf[i] = suf[i + 1] * nums[i + 1];
for(int i = 0; i < nums.size(); ++i)
result.emplace_back(pre[i] * suf[i]);
return result;
}
};
看了下灵神的题解,他的思路和我一样,也是用前后缀来做的,这里就不再赘述了。