[Python] 코딩 기초 트레이닝 - 주사위게임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
2	2	2	2	2222
4	1	4	4	1681
6	3	3	6	27
2	5	2	6	30
6	4	2	5	2

 

입출력 예 설명

---

입출력 예 #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 합니다.

 

개인 코드

---

def solution(a, b, c, d):
    answer = 0    
    element_list = [a,b,c,d]
    element_set = set(element_list)
        
    if len(element_set) == 1 :
        answer = a * 1111        
        
    elif len(element_set) == 4 :        
        answer = min(element_list)              
        
    elif len(element_set) == 3 :  
        
        # 어렵다
        # 요소가 리스트에 몇 개 있는지 count 하는 함수가 필요한가?
        new_element_list = []
        
        for x in element_list :            
            if element_list.count(x) != 2 :                 
                #중복 요소 제거하는 함수가 뭐지?
                new_element_list.append(x)
                                
        answer = new_element_list[0] * new_element_list[1]
        
    elif len(element_set) == 2 :      
        
            element_set_list = list(element_set)
            
            if (element_list.count(element_set_list[0]) == 3) :
                #각 요소 더하기, 덧셈하는 함수 기억이 안난다 ㅠㅠ                          
                answer = pow(10 * element_set_list[0] + element_set_list[1], 2)                
                          
            if (element_list.count(list(element_set)[0]) == 1) :
                answer = pow(10 * element_set_list[1] + element_set_list[0], 2)
              
            if element_list.count(element_list[0]) == 2 :
                # 와... ㅠㅠㅠㅠ 힘들어                
                answer = (element_set_list[0] + element_set_list[1] ) * (abs((element_set_list[0] - element_set_list[1] )) )
               
                
              
        
        
        
    
    
    
    return answer

 

---