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

문제설명

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

• 세 숫자가 모두 다르다면 a + b + c 점을 얻습니다.
• 세 숫자 중 어느 두 숫자는 같고 나머지
  다른 숫자는 다르다면 (a + b + c) × (a2 + b2 + c2 )점을 얻습니다.
• 세 숫자가 모두 같다면 (a + b + c) × (a2 + b2 + c2 ) × (a3 + b3 + c3 )점을 얻습니다.

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

제한사항

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

입출력 예

a	b	c	result
2	6	1	9
5	3	3	473
4	4	4	110592

풀이

설명에 나와있는 문제를 기준으로 if문을 작성하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>

using namespace std;

int solution(int a, int d, vector<bool> included) {
    int answer = 0;
    int num = 0;
    
    for(int i = 0; i < included.size(); i++)
    {
        num = a + (d * i);
        
        if(included[i])
            answer += num;
    }
    
    return answer;
}

결과