[프로그래머스] 연속 부분 수열 합의 개수

bej_ve·2022년 10월 27일
0

python알고리즘

목록 보기
37/46

<문제>
원형 수열의 모든 원소 elements가 순서대로 주어질 때, 원형 수열의 연속 부분 수열 합으로 만들 수 있는 수의 개수를 return 하도록 solution 함수를 완성해주세요.

elements result
[7,9,1,1,4] 18

<코드>

def solution(elements):
    answer = 0
    numberSet=set()
    
    n=len(elements)
    elements=elements*2
    for i in range(n):
        for j in range(n):
            numberSet.add(sum(elements[j:j+i+1]))

    answer=len(numberSet)
    return answer

<풀이>
set을 사용해서 중복되는 값을 자동으로 제거해준다.
list에서도 sum함수를 사용할 수 있다는 점!!

0개의 댓글