본문 바로가기
코딩테스트/파이썬

[Python/프로그래머스]코딩테스트 연속된 수의 합

by wyatti 2023. 6. 24.

연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num과 total이 주어집니다. 연속된 수 num개를 더한 값이 total이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.
파이썬 코딩테스트 연속된 수의 합

문제 설명

연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num과 total이 주어집니다. 연속된 수 num개를 더한 값이 total이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.

 

 

 

제한사항

  • 1 ≤ num ≤ 100
  • 0 ≤ total ≤ 1000
  • num개의 연속된 수를 더하여 total이 될 수 없는 테스트 케이스는 없습니다.

 

 

파이썬 프로그래머스 연속된 수의 합

def solution(num, total):
    answer = []
    start = total // num - (num - 1) // 2  # 시작 정수 계산

    for i in range(num):
        answer.append(start + i)

    return answer

위의 코드는 answer 리스트를 사용하여 정수 배열을 생성하는 방식으로 작성되었습니다. start 변수를 통해 시작 정수를 계산하고, 반복문을 통해 start부터 num개의 연속된 정수를 answer 리스트에 추가합니다.

댓글