[코딩 기초 문제] 홀짝에 따라 다른 값 반환하기

[코딩 기초 문제] 홀짝에 따라 다른 값 반환하기

 

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

 

 

(1) 먼저 정수 n이 짝수인지 홀수인지를 판별한다.

(2) 홀수라면-> 0~n 사이에 존재하는 홀수들을 모두 answer에 담아 반환한다.

(3) 짝수라면-> 0~n 사이에 존재하는 짝수들을 모두 제곱해, answer에 담아 반환한다!

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