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

📌  문제

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

---

제한사항

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

 

💡  아이디어

num이 짝수면, total을 num으로 나눈 값이 연속된 수를 반으로 나눈 바로 앞의 값이 된다.

num=6, total=21이면 (1, 2, 3, 4, 5, 6) 여기서 3이되는 것

num이 홀수면, total을 num으로 나눈 값(나머지 버리고)이 연속된 수의 중간 값이 된다.

이를 이용해서 middle을 구해놓고, for문을 어디서부터 어디까지 돌릴지 생각해본다.

 

✏️  풀이

function solution(num, total) {
  var answer = [];

  if(num % 2 === 0){
    let middle = Math.floor(total / num);
    let from = middle - (num/2 - 1);	// 첫번째 수는 num을 2로 나누고, 1을 뺀 값을 middle에서 뺀 값
    let to = middle + (num/2);	// 마지막 수는 middle에 num/2한 값을 더한 값
    for(let i=from; i<=to; i++){
      answer.push(i);
    }
  }
  else{
    let middle = Math.floor(total / num);
    let from = middle - (Math.floor(num/2));
    let to = middle + (Math.floor(num/2));
    for(let i=from; i<=to; i++){
      answer.push(i);
    }
  }

  return answer;
}