LeetCode238. Product of Array Except Self

最新推荐文章于 2024-06-12 07:12:52 发布

原创最新推荐文章于 2024-06-12 07:12:52 发布 · 116 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

503 篇文章

订阅专栏

博客围绕给定数组，要求返回一个数组，其元素为原数组除自身外所有元素的乘积。解题难点在于处理0元素，分无0元素、1个0元素、2个及以上0元素三种情况讨论，并给出了相应思路，还提及不使用除法且时间复杂度为O(n)等要求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given an array nums of n integers where n > 1, return an array output such that output[i]is equal to the product of all the elements of nums except nums[i].

Example:

Input:  [1,2,3,4]
Output: [24,12,8,6]

Note: Please solve it without division and in O(n).

Follow up:
Could you solve it with constant space complexity? (The output array does not count as extra space for the purpose of space complexity analysis.)

思路：

这道题我开始以为会有非常大的数出现作为一个难点，但是并没有；这道题的难点在于解决0元素的问题，我们这里分三种情况讨论：

1 没有0元素，这个时候我们只需要按照常规的用累乘除以目标数就可以了

2 有1一个0元素，这个时候除了0元素的位置，其他的数应该都是0

3 有2个及以上的0元素，这个时候所有位置的元素最后都会变成0

代码如下：

class Solution {
public:
    vector<int> productExceptSelf(vector<int>& nums) {
		int product = 1;
		int zeroCount = 0;
		for (int num : nums) {
			if (num == 0) zeroCount++;
			else product *= num;
		}
		if (zeroCount > 1) return vector<int>(nums.size(), 0);

		if (zeroCount > 0) {
			for (int& num : nums) {
				if (num == 0) num = product;
				else num = 0;
			}
		}
		else {
			for (int& num : nums) num = product / num;
		}
		return nums;
    }
};