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

최초 풀이 - 오답, 예시 케이스 통과

def solution(elements):
    answer = []
    for i in range(len(elements)):
        count = 0
        tmp = []
        for j in range((i + 1) * len(elements)):
            j = (j % len(elements))
            tmp.append(elements[j])
            
            if count == i:
                print(sum(tmp))
                answer.append(sum(tmp))
                tmp = []
                count = 0
            else:
                count += 1
            
        print()
        
    return len(list(set(answer)))

 

정답 풀이

def solution(elements):
    result = set()
    for i in range(len(elements)):
        for j in range(len(elements)):
            result.add(sum((elements * 2)[j : j + i + 1]))
    return len(result)