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

문제

홀짝에 따라 다른 값 반환하기 : 문제 링크

문제 분석

• 양의 정수 n이 매개변수로 주어질 때, n이 홀수라면 n이하의 홀수인 모든 양의 정수의 합을 return하고, n이 짝수라면 n이하의 짝수인 모든 양의 정수의 제곱의 합을 return
• if ~ else문을 통해 매개변수 n이 홀수, 짝수인지 판별. 홀수일경우 for loop를 통해 1부터 n까지 검사하여 홀수 정수일경우 값을 answer에 누적하여 더하고, 짝수일경우 for loop를 통해 2부터 n까지 검사하여 짝수 정수일경우 값을 제곱하여 answer에 누적하여 더함. 최종적으로 저장된 answer을 return

풀이

using namespace std;

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