[프로그래머스/파이썬] 주사위 게임 3(Lv.0)

#미션
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 함수를 작성해 주세요.

#내답

def solution(a, b, c, d):
    ls = [a,b,c,d]
    dic = {i : ls.count(i) for i in set(ls)}
    num = len(set(ls))
    sorted_dic = sorted(dic.items(),key=lambda x:x[1], reverse = True)

    if num == 1:
        return 1111*a
    elif num == 4:
        return min(ls)
    elif num == 2:
        p = sorted_dic[0][0] 
        q = sorted_dic[1][0]
        if sorted_dic[0][1] == 3:
            return (10*p + q)**2
        else:
            return (p+q) * abs(p-q)
    else:
        p = sorted_dic[0][0] 
        q = sorted_dic[1][0]
        r = sorted_dic[2][0]
        return q*r

#수정한답
def solution(a, b, c, d):
    nums = [a, b, c, d]
    counts = [nums.count(i) for i in nums]
    
    if max(counts) == 4:
        return 1111 * a
    elif max(counts) == 3:
        p = nums[counts.index(3)]
        q = nums[counts.index(1)]
        return (10*p + q)**2
    elif max(counts) == 2:
        if min(counts) == 2:
            return abs((a+b)*(a-b)) if a!=b else abs((a+c)*(a-c))
        else:
            p = nums[counts.index(2)]
            return (a*b*c*d)/p**2
    else: 
        return min(nums)

수정한 답 제대로 적어내기까지 몇시간 걸렸다;;

일단 nums,counts로 if 시작전에 내가 적어낸 4줄보다 반이 줄어서 2줄로 해결함.

max(counts)로 개수 세는거 해결, 그리고 필요시 min(counts) 에다가 마지막쯤 (a*b*c*d)/p**2은 아름다울 지경;;

분발해야겠다!!🔥