题目
Given an array of n integers where n > 1, nums, return an array output such that output[i] is equal to the product of all the elements of nums except nums[i].
Solve it without division and in O(n).
For example, given [1,2,3,4], return [24,12,8,6].
Follow up:
Could you solve it with constant space complexity? (Note: The output array does not count as extra space for the purpose of space complexity analysis.)
翻译
给定一个数组A,具有n个元素,n大于1,返回一个数组,数组元素output[i]为,除过元素A【i】,其他元素的乘积。时间复杂度为O(n)
注:
是否可以用O(1)的空间复杂度结果问题,
思路
首先求出数组的成绩,然后对每个元素,除以它,就得到了其他元素乘积的结果,这样子,时间复杂度就是O(n)
但是这里边要注意坑,就是如果里边有0的情况。
这里我们分析得到:
1,如果成绩为0,则数组中一定有数位0.
2,如果数组有由两个以上的数据位0,则所有的结果均为0.
3,如果只有一个,则这个0处的结果不为0外,其他均为0.
代码
class Solution {
public:
vector<int> productExceptSelf(vector<int>& nums) {
int all = 1;
int cnt_zero = 0;
vector<int> res;
for(int i = 0; i < nums.size();i++){
if(nums[i] == 0){
cnt_zero++;
}
all *= nums[i];
}
//成绩为0,说明其中有0,如果0的个数多余1个,则所有都是0,
//如果只有一个,则除了0元素非零外,其他元素均为0
if(all == 0){
if(cnt_zero > 1){
res = vector<int>(nums.size(),0);
}else{
res = vector<int>(nums.size(),0);
int pos = -1;
for(int i = 0; i < nums.size();i++){
if(nums[i] == 0){
pos = i;
}
}
res[pos] = 1;
for(int i = 0; i < nums.size(); i++){
if(i != pos){
res[pos] *= nums[i];
}
}
}
}else{
for(int i = 0; i < nums.size(); i++){
res.push_back(all/nums[i]);
}
}
return res;
}
};