每日一题,防止痴呆 = =
一、题目大意
给你一个长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/product-of-array-except-self
二、题目思路以及AC代码
好像偶尔有几天没有做题,这习惯不太好,以后要坚持!!!
思路:前缀数组、后缀数组
这道题其实思路很明显,要求取除该元素以外所有元素的乘积,无非就是求该元素前面元素的乘积乘该元素后面元素的乘积嘛,这样我们只要维护前缀数组和后缀数组就可以在O(n)复杂度内进行计算了。
但题目后来要求我们只用常数空间复杂度,而且括号里说输出数据的空间不算,那很自然想到利用输出的这部分空间计算其中的一个数组(前缀or后缀),然后另一个数组可以利用一个变量动态生成,也可以利用nums去存储,反正占用的空间都是常数级的。
AC代码
空间复杂度为O(n),时间复杂度为O(n)
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
vector<int> res;
int n_size = nums.size();
if (!n_size) return res;
int* pre = new int[n_size];
int* pro = new int[n_size];
pre[0] = 1;
for (int i=1;i<n_size;i++) {
pre[i] = pre[i-1] * nums[i-1];
}
pro[n_size-1] = 1;
for (int i=n_size-2;i>=0;i--) {
pro[i] = pro[i+1] * nums[i+1];
}
for (int i=0;i<n_size;i++) {
res.push_back(pre[i] * pro[i]);
}
return res;
}
};
空间复杂度为O(1),时间复杂度为O(n)
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int n_size = nums.size();
vector<int> res(n_size);
if (!n_size) return res;
res[n_size-1] = 1;
for (int i=n_size-2;i>=0;i--) {
res[i] = res[i+1] * nums[i+1];
}
int last = nums[0];
nums[0] = 1;
for (int i=1;i<n_size;i++) {
int tmp = last * nums[i-1];
last = nums[i];
nums[i] = tmp;
}
for (int i=0;i<n_size;i++) {
res[i] *= nums[i];
}
return res;
}
};
如果有问题,欢迎大家指正!!!