4일차 / 홀짝에 따라 다른 값 반환하기 / C++ / 기초

홀짝에 따라 다른 값 반환하기

문제 설명

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

제한사항

• 1 ≤ n ≤ 100

 

입출력 예

n	result
7	16
10	220

 

 

입출력 예 설명

입출력 예 #1

예제 1번의 n은 7로 홀수입니다. 7 이하의 모든 양의 홀수는 1, 3, 5, 7이고 이들의 합인 1 + 3 + 5 + 7 = 16을 return 합니다.

 

입출력 예 #2

예제 2번의 n은 10으로 짝수입니다. 10 이하의 모든 양의 짝수는 2, 4, 6, 8, 10이고 이들의 제곱의 합인 22 + 42 + 62 + 82
+ 102 = 4 + 16 + 36 + 64 + 100 = 220을 return 합니다.

 

 

---

 

내 코드

#include <string>
#include <vector>

using namespace std;

int solution(int n) {
    int answer = 0;
    if ( n % 2 == 0 ) {
        while ( n ){
            if ( n % 2 == 0 )
                answer += n * n;
            n--;
        }
    }
    else
        while ( n ) {
            if ( n % 2 != 0 )  
                answer += n;
            n--;
        }
    return answer;
}

• 쓸데없이 긴 코드
• 조금 더 간략화 필요

 

 

다른 사람의 코드

#include <string>
#include <vector>

using namespace std;

int solution(int n) {
    int answer = 0;
    while(n>0)
    {
        answer += n%2==0 ? n*n : n;
        n-=2;
    }
    return answer;
}

• 이것과 비슷한 코드를 생각했는데 오류가 나서 더 고민하지 않고 쉬운 길을 택했다
• 간략한 코드에 대한 끈기 있는 고민이 필요할 듯