[프로그래머스] 18. 홀짝에 따라 다른 값 반환하기

입출력 예

입출력 예 설명

• 예제 1번의 n은 7로 홀수입니다. 7 이하의 모든 양의 홀수는 1, 3, 5, 7이고 이들의 합인 1 + 3 + 5 + 7 = 16을 return 합니다.

• 예제 2번의 n은 10으로 짝수입니다. 10 이하의 모든 양의 짝수는 2, 4, 6, 8, 10이고 이들의 제곱의 합인 2 + 4 + 6 + 8 + 10 = 4 + 16 + 36 + 64 + 100 = 220을 return 합니다.

코드

class Solution {
    public int solution(int n) {
        int result = 0;
        if(n % 2 != 0) {
            for (int i = 1; i <= n; i = i + 2) {
                result += i;
            }
        } else {
            for (int i = 2; i <= n; i = i + 2) {
                result += i * i;
            }
        }
        return result;
    }
}

수학 공식을 이용한 코드 (공식 나중에 찾아보기)

class Solution {
    public int solution(int n) {
        if (n % 2 == 1) {
            return (n + 1) * (n + 1) / 2 / 2;
        } else {
            return 4 * n/2 * (n/2 + 1) * (2 * n/2 + 1) / 6;   
        }
    }
}