LeetCode—Python—238. 除自身以外数组的乘积

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/IOT_victor/article/details/88136265

1题目描述

给定长度为 n 的整数数组 nums,其中 n > 1,返回输出数组 output ,其中 output[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积。

示例:

输入: [1,2,3,4]
输出: [24,12,8,6]

说明: 不要使用除法,且在 O(n) 时间复杂度内完成此题。

进阶:
你可以在常数空间复杂度内完成这个题目吗?( 出于对空间复杂度分析的目的,输出数组不被视为额外空间。)

2解决思路

3代码实现

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        #从左往右
        output = [1] * len(nums)
        for i, num in enumerate(nums[:-1]):
            output[i + 1] = output[i] * num
        
        #从右往左遍历
        right = 1
        for i in range(len(nums) - 1, -1, -1):
            #乘上(从左往右时)未乘项
            output[i] = output[i] * right
            right = right * nums[i]
        
        return output

4补充

列表对应位置相乘

left = [1,1,2,6]
right = [24,12,4,1]
# c = left * right #错误,列表不能相乘,可用numpy或zip()


# 法一
# c = []
# for i in range(len(left)):
#     for j in range(i, i + 1): #j = i
#         c.append(left[i] * right[j])

# 法二   
c = [a * b for a, b in zip(left, right)]

# 法三
a = np.array([1,1,2,6])
b = np.array([24,12,4,1])
c = a * b

猜你喜欢

转载自blog.csdn.net/IOT_victor/article/details/88136265