[프로그래머스][LV.0] 주사위 게임 3 | python3

 

문제 링크: 주사위 게임 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 이하의 정수입니다.

 

---

문제 풀이

 

원래 엄청 조건을 나눠서 풀었는데...분명 더 좋은 방법이 있을 것 같아서 문제를 따로 빼놨었다.
그리고 나는 시도를 하고 정렬하게 실패...

def solution(a, b, c, d):
    answer = 0
    chk={}
    for i in [a,b,c,d]:
        chk[i]+=1
    if 4 in chk:
        answer=1111*chk.index(4)
    elif 3 in chk:
        answer=(10*chk.index(3)+chk.index(1))**2
    elif chk.count(2)==2:
        p=chk.index(2)
        chk.remove(2)
        q=chk.index(2)+1
        answer=(p+q)*abs(p-q)
    elif chk.count(0)==4:
        
    else
    return answer​​

이런식으로 리스트를 활용하여 풀려다가 정혀 코드가 짧아질 것 같지 않아서 딕셔너리로 변경....해서 풀다가 딕셔너리도 뭔가 똑같이 복잡하고 길게 될 것 같아서 일단 스톱했다.

 

내코드

def solution(a, b, c, d):
    answer = 1
    m=sorted([a,b,c,d])
    l=len(set(m))
    
    if l==1:
        answer=1111*a
    elif l==2:
        if m[0]==m[1] and m[2]==m[3]:
            answer=(m[0]+m[2])*abs(m[0]-m[2])
        else:
            if m[0]==m[2]:
                answer=(10*m[0]+m[3])**2
            else:
                answer=(10*m[3]+m[0])**2
    elif l==3:
        for n in range(1,7):
            if [a,b,c,d].count(n) == 2:
                break
        for q in [a,b,c,d]:
            if q != n:
                answer*=q
    else:
        answer=min(a,b,c,d)
    return answer

 

 

---

다른사람풀이

def solution(a, b, c, d):
    l = [a,b,c,d]
    c = [l.count(x) for x in l]
    if max(c) == 4:
        return 1111*a
    elif max(c) == 3:
        return (10*l[c.index(3)]+l[c.index(1)])**2
    elif max(c) == 2:
        if min(c) == 1:
            return eval('*'.join([str(l[i]) for i, x in enumerate(c) if x == 1]))
        else:
            return (max(l) + min(l)) * abs(max(l) - min(l))
    else:
        return min(l)

 

def solution(a, b, c, d):
    nums = [a, b, c, d]
    counts = [nums.count(i) for i in nums]
    if max(counts) == 4:
        return a * 1111
    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 (a + c) * abs(a - c) if a == b else (a + b) * abs(a - b)
        else:
            p = nums[counts.index(2)]
            return (a * b * c * d) / p**2
    else:
        return min(nums)

같은 리스트인데 이렇게 길이가 다를수가 있나...

max와 min 을 잘 활용하여 간단하게 푼 풀이 같다. 리스트 인덱스를 주사위 숫자로 넣고 나오는 횟수만큼 원소로 더한다는 생각은 같았는데.....max가 2일때 if else의 조건이 정말 깔끔한 것 같다.