연속된 세 개의 정수를 더해 12가 되는 경우는 3, 4, 5입니다. 두 정수 num
과 total
이 주어집니다. 연속된 수 num
개를 더한 값이 total
이 될 때, 정수 배열을 오름차순으로 담아 return하도록 solution함수를 완성해보세요.
num
≤ 100total
≤ 1000num
개의 연속된 수를 더하여 total
이 될 수 없는 테스트 케이스는 없습니다.num | total | result |
---|---|---|
3 | 12 | [3, 4, 5] |
5 | 15 | [1, 2, 3, 4, 5] |
4 | 14 | [2, 3, 4, 5] |
5 | 5 | [-1, 0, 1, 2, 3] |
def solution(num, total):
answer = []
if total % num == 0:
subNum = num // 2
else:
subNum = num // 2 - 1
firstNum = (total // num) - subNum #result arr 시작 숫자
for i in range(num):
answer.append(firstNum+i)
return answer
구현 문제
answer
배열의 시작 숫자를 어떻게 잡을 수 있을지 고민해보는 문제
2가지 상황을 고려 후, 입출력 예의 테스트케이스를 분석 시, 규칙이 보이는 문제