[프로그래머스/C++ 문제 풀이] Lv. 0 - 주사위 게임 3

문제 설명

1부터 6까지 숫자가 적힌 주사위가 네 개 있습니다. 네 주사위를 굴렸을 때 나온 숫자에 따라 다음과 같은 점수를 얻습니다.

• 네 주사위에서 나온 숫자가 모두 p로 같다면 1111 × p점을 얻습니다.
• 세 주사위에서 나온 숫자가 p로 같고 나머지 다른 주사위에서 나온 숫자가 q(p ≠ q)라면 (10 × p + q)2 점을 얻습니다.
• 주사위가 두 개씩 같은 값이 나오고, 나온 숫자를 각각 p, q(p ≠ q)라고 한다면 (p + q) × |p - q|점을 얻습니다.
• 어느 두 주사위에서 나온 숫자가 p로 같고 나머지 두 주사위에서 나온 숫자가 각각 p와 다른 q, r(q ≠ r)이라면 q × r점을 얻습니다.
• 네 주사위에 적힌 숫자가 모두 다르다면 나온 숫자 중 가장 작은 숫자 만큼의 점수를 얻습니다.

네 주사위를 굴렸을 때 나온 숫자가 정수 매개변수 a, b, c, d로 주어질 때, 얻는 점수를 return 하는 solution 함수를 작성해 주세요.

제한 사항

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

입출력 예

a	b	c	d	result
4	1	4	4	1681
6	3	3	6	27
2	5	2	6	30
6	4	2	5	2
2	2	2	2	2222

 

입출력 예 #1

• 예제 1번에서 네 주사위 숫자가 모두 2로 같으므로 1111 × 2 = 2222점을 얻습니다. 따라서 2222를 return 합니다.

입출력 예 #2

• 예제 2번에서 세 주사위에서 나온 숫자가 4로 같고 나머지 다른 주사위에서 나온 숫자가 1이므로 (10 × 4 + 1)2 = 412 = 1681점을 얻습니다. 따라서 1681을 return 합니다.

입출력 예 #3

• 예제 3번에서 a, d는 6으로, b, c는 3으로 각각 같으므로 (6 + 3) × |6 - 3| = 9 × 3 = 27점을 얻습니다. 따라서 27을 return 합니다.

입출력 예 #4

• 예제 4번에서 두 주사위에서 2가 나오고 나머지 다른 두 주사위에서 각각 5, 6이 나왔으므로 5 × 6 = 30점을 얻습니다. 따라서 30을 return 합니다.

입출력 예 #5

• 예제 5번에서 네 주사위 숫자가 모두 다르고 나온 숫자 중 가장 작은 숫자가 2이므로 2점을 얻습니다. 따라서 2를 return 합니다.

---

문제 풀이

이번 문제는 저의 경우 Map 컨테이너를 활용해 현재 Map의 사이즈를 조사하여 사이즈 별로 경우의 수를 찾아 문제를 해결하였습니다.

#include <string>
#include <vector>
#include <map>

using namespace std;

int solution(int a, int b, int c, int d) {
    int answer = 0;

    vector<int> nums = {a, b, c, d};
    map<int, int> m;

    // 맵에 숫자 넣기
    for (int num : nums)
    {
        // 맵에 해당 숫자의 키가 존재하지 않는다면?
        if (m.find(num) == m.end())
        {
            // 추가
            m.insert(make_pair(num, 1));
        }
        // 존재한다면?
        else
        {
            // 해당 키의 값 증가
            m[num]++;
        }
    }

    // 만약 맵의 사이즈가 1이라면? (모두 같은 숫자)
    if (m.size() == 1)
    {
        answer = 1111 * m.begin()->first;
    }
    // 만약 맵의 사이즈가 2라면? (3개 같은 숫자, 남은 1개 다른 숫자 or 2개씩 같은 숫자)
    else if (m.size() == 2)
    {
        int p = 0;
        int q = 0;

        bool IsDoubleNum = false;

        for (pair<int, int> num : m)
        {
            // 3 : 1 비율이라면?
            if (num.second == 3 || num.second == 1)
            {
                if (num.second == 1)
                {
                    q = num.first;
                }
                else
                {
                    p = num.first;
                }
            }
            // 2 : 2 비율이라면?
            else
            {
                IsDoubleNum = true;

                if (p == 0)
                {
                    p = num.first;
                }
                else
                {
                    q = num.first;
                }
            }

        }

        if (IsDoubleNum)
        {
            answer = (p + q) * abs(p - q);
        }
        else
        {
            answer = (10 * p + q) * (10 * p + q);
        }
    }
    // 만약 맵의 사이즈가 3이라면? (2개 같은 수 2개 다른 수)
    else if (m.size() == 3)
    {
        int q = 0;
        int r = 0;

        for (pair<int, int> num : m)
        {
           if (num.second == 1)
           {
               if (q == 0)
               {
                   q = num.first;
               }
               else
               {
                   r = num.first;
               }
           }
        }

        answer = q * r;
    }
    // 모두 다른 숫자라면?
    else if (m.size() == 4)
    {
        // 가장 작은 숫자(맵의 첫 번째 요소) 지정하기
        answer = m.begin()->first;
    }

    return answer;
}