[Programmers/C++] 주사위 게임 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 함수를 작성해 주세요.

 

제한 사항

 

 

 

 

 

입출력 예시

입출력 예시

 

입출력 예시

 

 

 

---

 

풀이 

#include <vector>
#include <set>
#include <map>
#include <math.h>
using namespace std;

int solution(int a, int b, int c, int d) {
    
    set<int> nums{a, b, c, d};
    map<int, int> count;
    
    count[a]++;
    count[b]++;
    count[c]++;
    count[d]++;
    
    switch(nums.size()){
        
        //. 네 주사위에서 나온 숫자가 모두 같을 때 (p p p p)
        case 1: 
            return 1111 * a;
            
        case 2: 
            //. 주사위가 두 개씩 같은 값일때 (p p q q)
            if (count[a] == 2 || count[b] == 2 || count[c] == 2 || count[d] == 2){
                auto it = nums.begin();
                int p = *it;
                advance(it, 1);
                int q = *it;
                
                return (p + q) * abs(p - q);
            }
            
            //. 세 주사위에서 나온 숫자가 같고 나머지 주사위 숫자만 다를 때 (p p p q)
            else {
                int p, q;
                for (const auto& c : count){
                    if (c.second == 3)
                        p = c.first;
                    else
                        q = c.first;
                }
                    
                return pow(10 * p + q, 2);
            }
            
        //. 두 수만 같고 나머지는 다를 때 (x x y z)
        case 3: {
            vector<int> v;
            for (const auto& c : count){
                if (c.second != 2) 
                    v.push_back(c.first);
            }
            return v[0] * v[1];
        }
        
        case 4:
            auto it = nums.begin();
            return *it;
    }
}

 

 

 

---