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

 

<1st Solution> 

class Solution {
    public int solution(int n) {
        int answer = 0;
        
        if( n % 2 == 1) {
            answer = odd(n);
        } else {
            answer = even(n);
        }
        
        return answer;
    }
    
    // 홀수
    public int odd(int n) {
        int k = 0;
        
        for( int i = 1; i <= n; i++) {
            if( i % 2 == 1) {
                k += i;
            }
        }
        return k;
    }
    
    // 짝수
    public int even(int n) {
        int k = 0;
        
        for( int i = 1; i <= n; i++) {
            if( i % 2 == 0) {
                k += i*i;
            }
        }
        return k;
    }
}

 

아무생각 없이 Method를 만들어서 문제를 풀었다.

그래서 Method 없이 짧게 써보려고 다시 시도했다.

 

<2nd Solution> 

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;
    }
}