[Python] 연속 부분 수열 합의 개수

입출력 예

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

 

문제 풀이

아하... length로 해야지 딱 주어진 [7, 9, 1, 1, 4]으로 한 합이 된다!

 

def solution(elements):
    result = set()
    length = len(elements)
    elements = elements * 2
    # [7, 9, 1, 1, 4, 7, 9, 1, 1, 4]
    for i in range(length): #  0 1 2 3 4
        for j in range(length): # 0 1 2 3 4
            result.add(sum(elements[j:j+i+1]))
    return len(result)