인천의 자유인

[Python] 연속된 수의 합 - 프로그래머스 본문

Python/Python코딩테스트

[Python] 연속된 수의 합 - 프로그래머스

Youngook 2024. 5. 13. 09:56
728x90
반응형

 

 

 

나의 문제 풀이

def solution(num, total):
    answer = []
    x = -num
    while True:
        total1 = total
        for i in range(x,x+num):
            total1 = total1 -i
        if total1 == 0:
            for j in range(x, x+num):
                answer.append(j)
            break
        x+=1
    return answer

 

먼저 x 변수는 -num으로 만듭니다. 그 이유는 원소 안에 -가 들어 있는 경우가 있고 -total로 하기에는 너무 루프를 많이 돌리게 되어서 가장 적당하게 -num으로 했습니다.

그 후에 total1을 total로 만들어서 x,x+1,x+2...x+num을 차례대로 뺍니다. 이렇게 빼서 0이 되는 x를 발견하게 되면 연속된 수의 합을 구할 수 있습니다.(x는 지속적으로 1씩 증가해서 모든 경우의 수를 검색할 수 있음)

예를 들어 num이 3이고 12가 total이었을 때 12 -3 -4 -5를 하게 되면 0이 됨으로 그 뺀 수들을 리스트에 추가해서 [3,4,5]를 만들수 있게 됩니다. 그리고 그 리스트가 완성이 될 경우에는 return을 하면 문제 풀이 완성이 됩니다.

728x90
반응형