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

문제설명

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

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

두 정수 a와 b가 매개변수로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

제한사항

• a와 b는 1 이상 6 이하의 정수입니다.

입출력 예

a	b	result
3	5	34
6	1	14
2	4	2

풀이

주사위 a, b의 숫자를 기준으로 조건에 맞게 로직을 구현하는 간단한 구현 문제입니다.

#include <string>
#include <vector>

using namespace std;

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

결과