[LeetCode] Product of Array Expect Self

[LeetCode] Product of Array Expect Self

 

1. 문제 : Link

 

2. 풀이

0의 유무와 개수에 따라 나눠서 값을 넣어준다.

총 합에서 나누기를 하면 O(n)을 유지할 수 있음!

 

3. 코드

class Solution:
    def productExceptSelf(self, nums: List[int]) -> List[int]:
        tmp = 1
        check = 0
        arr = []
        for n in nums:
            if n == 0:
                check += 1
                continue
            else:
                tmp *= n
        if check >= 2:
            for _ in range(len(nums)):
                arr.append(0)
        elif check == 1:
            for n in nums:
                if n == 0:
                    arr.append(tmp)
                else:
                    arr.append(0)
        else:
            for n in nums:
                arr.append(tmp // n)
        return arr