[프로그래머스 / C++] 주사위 게임 1

문제

주사위 게임 1 : 문제 링크

문제 분석

• 1부터 6까지 숫자가 적힌 주사위가 두개 있다. 두 주사위를 굴렸을 때 나온 숫자를 각각 a, b라고 했을 때 얻는 점수는 다음과 같다.

  • a와 b가 모두 홀수라면 a^2 + b^2 점을 얻는다.
  • a와 b중 하나만 홀수라면 2 x (a + b)점을 얻는다.
  • a와 b 모두 홀수가 아니라면 | a - b | 점을 얻는다.

• 두 정수 a와 b가 매개변수로 주어질 때, 얻는 점수를 return
• 정수를 절대값으로 만드는 abs() 함수를 사용하기 위해 cstdlib 헤더를 include
• if ~ else if ~ else문을 통해 각 조건을 설정하고, 정수형 변수 answer에 매개변수 a, b를 활용하여 연산한 결과를 저장. a, b 모두 홀수가 아닐경우, 즉 둘다 짝수일경우에는 abs() 함수를 활용하여 a - b 값을 절대값으로 치환. 최종적으로 저장된 answer를 return

풀이

#include <cstdlib>

using namespace std;

int solution(int a, int b) {
    int answer = 0;
    
    if(a % 2 == 1 && b % 2 == 1) answer = (a * a) + (b * b);
    else if((a % 2 == 1 && b % 2 == 0) || (a % 2 == 0 && b % 2 == 1)) answer = 2 * (a + b);
    else answer = abs(a - b);
    return answer;
}