프로그래머스 - 주사위 게임 3 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/181916

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

int solution(int a, int b, int c, int d) {
    int answer = 0;
    int ans[6]={0,};
    ans[a-1]++;
    ans[b-1]++;
    ans[c-1]++;
    ans[d-1]++;
    int max=-1;
    int maxId=-1;
    int min=7;
    
    for(int i=0;i<6;i++){
        if(max<ans[i]){
            max=ans[i];maxId=i;
        }
        
    }
    if(max==4)return 1111*(maxId+1);
    else if(max==3)
    {
        for(int i=0;i<6;i++)
        {
            if(ans[i]==1)
            {
                return (10*(maxId+1)+(i+1))*(10*(maxId+1)+(i+1));
            }
        }
    }
    else if(max==2)
    {
        for(int i=0;i<6;i++)
        {
             if(ans[i]==2&&i!=maxId)
             {
                 return ((maxId+1)+(i+1))*( (maxId>i)?(maxId-i):(i-maxId) );
             }
             if(ans[i]==1)
             {
                 for(int j=0;j<6;j++)
                 {
                     if(j!=i&&ans[j]==1)return (j+1)*(i+1);
                 }
             }
        }
    }
    else 
    {
        for(int i=0;i<6;i++)
        {
            if(ans[i]==1)return i+1;
        }
    }
    return answer;
}

설명 

조건문만 나열하는 방식을 벗어난 방법을 사용해보고 싶어서 ans 배열에 나온 숫자의 개수를 저장하는 방식을 사용했다. 

가장 많이 나온 숫자는 max와 maxId에 저장이 된다. max는 가장많이 나온 숫자가 몇개인지, maxId는 그 숫자가 몇인지를 저장한다. 

그리고 max에 따라 문제에서 주어진 출력 값을 계산하여 return 한다. 

 

조건문 나열보다 간단하게 코드를 작성해보려고 했는데 각 조건일 때마다 for문을 돌며 원하는 값을 찾아야 해서 오히려 코드가 복잡해져 가독서이 떨어지는 듯하다.