홀짝에 따라 다른 값 반환하기

TIS Day 18

홀짝에 따라 다른 값 반환하기

양의 정수 n이 매개변수로 주어질 때, n이 홀수라면 n 이하의 홀수인 모든 양의 정수의 합을 return 하고 n이 짝수라면 n 이하의 짝수인 모든 양의 정수의 제곱의 합을 return 하는 solution 함수를 작성해 주세요.

• 제한 사항
  1 ≤ n ≤ 100
• 입출력 예
  

풀이

function solution(n) {
    var answer = 0;
    var even = [];
    var odd = [];
    
    for (let i = 1; i < n + 1 ; i++){
        if ( n % 2 === 0 && i % 2 === 0) {
            even.push(i)
        } else if( n % 2 !== 0 && i % 2 !==0) {
            odd.push(i)
        }
    }
    
    const evenResult = even.length !== 0 ? even.reduce((accumulator, currentValue) => accumulator + currentValue ** 2, 0) : 0
    
    const oddResult = odd.length !== 0  ? odd.reduce((accumulator, currentValue) => accumulator + currentValue, 0) : 0

   
    return evenResult === 0 ? oddResult : evenResult;
}