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

문제 보기

https://school.programmers.co.kr/learn/courses/30/lessons/131701

프로그래머스

 

문제 해결

- 처음 해결한 풀이

시간초과가 날뻔해서 정답이었지만 바람직하지 않은 코드인 것 같다.

def solution(elements):
    answer = 0
    
    elements_2 = elements * 2
    j = 1
    sum_list = set()
    
    while j <= len(elements):
        for i in range(len(elements)):
            sum_list.add(sum(elements_2[i:i+j]))
        j += 1    
    return len(list(sum_list))

 

 

소스 코드 

- 다른 사람 풀이

def solution(elements):
    result = set()

    for i in range(len(elements)):
        tmp_sum = elements[i]
        result.add(tmp_sum)
        for j in range(i+1, i+len(elements)):
            tmp_sum += elements[j % len(elements)]
            result.add(tmp_sum)
    return len(result)