자바 알고리즘 - 홀짝에 따라 다른 값 반환하기

문제 설명

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

 

제한 사항

1 ≤ n ≤ 100

 

답안

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

 

n이 홀수 일 땐 for문을 돌면서 answer에 홀수값을 더해주고 n이 짝일땐 answer에 i*i 즉 i의 제곱을 더해준다.

이 때 for문의 증감식에 i++을 쓰는 것이 아닌 i+2를 사용해서 홀수 또는 짝수만을 반복하도록 한다.