[알고리즘] 자신을 제외한 배열의 곱

June·2021년 1월 17일
0

알고리즘

목록 보기
21/260

자신을 제외한 배열의 곱

내 풀이

문제를 보자마자 전체 배열을 다 곱한 수를 구한 다음, 각 배열의 값마다 나눠서 값을 저장하면 되겠다라고 생각했으나 문제 제약 사항에 나누기를 쓰지 말라고 되어있었다.

책 풀이

def productExceptSelf(nums: List[int]) -> List[int]:
    out = []
    p = 1

    for i in range(0, len(nums)):
        out.append(p)
        p = p * nums[i]

    p = 1
    for i in range(len(nums)-1, -1, -1):
        out[i] = out[i] * p
        p = p * nums[i]

    return out

0개의 댓글