[Python/파이썬] 프로그래머스 연속 부분 수열 합의 개수

프로그래머스

 

코드

def solution(elements):
    answer = set()
    new_arr = elements + elements
    for i in range(1, len(elements)+1): # 부분 수열의 길이
        start = 0
        while start <= len(elements):
            answer.add(sum(new_arr[start:start+i]))
            start += 1
    return len(answer)

원형 수열을 실제로 구현하지는 않았고 elements 수열을 2개 연달아 이어붙여서 부분 수열의 합을 구할 수 있도록 하였다. 부분 수열의 길이별로 시작점을 1씩 증가시키며 부분 수열의 합을 구하여 집합에 계속 추가하였다. 중복 제거를 하기 위하여 집합을 사용하였다. 그리고 최종적으로 집합의 길이를 구하면 정답이다.