프로그래머스 Java홀짝에 따라 다른 값 반환하기

 

 

 

 

 

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

 

제한사항
1 ≤ n ≤ 100

 

 

 

 

풀이

우선 홀수와 짝수를 구분하기 위해 if문을 사용해서 나머지 % 사용해 구분했습니다.

홀수일 경우는 n 이하의 모든 홀수의 값을 더해야 for문으로 반복문 사용해 하나씩 더해줬습니다.

짝수일 경우는 n 이하의 모든 짝수의 제곱을 더하기 때문에 Math.pow()를 사용했습니다.

 

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 j = 2; j <= n; j+=2){
            answer += Math.pow(j, 2);
          }
        }
        return answer;
    }
}