자기개발👨‍💻/코딩 알고리즘

[python] 백준 2480 주사위 세개

천숭이 2020. 12. 27. 20:40

www.acmicpc.net/problem/2480

 

2480번: 주사위 세개

1에서부터 6까지의 눈을 가진 3개의 주사위를 던져서 다음과 같은 규칙에 따라 상금을 받는 게임이 있다.  같은 눈이 3개가 나오면 10,000원+(같은 눈)*1,000원의 상금을 받게 된다.  같은 눈이 2개만

www.acmicpc.net

def same2(x,y):
    if x==y:
        return 1000+x*100

a,b,c=map(int,input().split())
if a==b and b==c:
    print(10000+a*1000)
elif a==b or a==c or b==c:
    if same2(a,b)!=None:print(same2(a,b))
    if same2(b,c)!=None:print(same2(b,c))
    if same2(a,c)!=None:print(same2(a,c))    
    
else:
    print(max(max(a,b),c)*100)

주사위 눈이 같은 것이 두개일때는 same2함수를 작성해 코드의 길이를 줄여주었다.

만약 인자로 받는 x와y의 수가 같지 않으면 return 값은 None이기 때문에 조건문을 사용해 출력문을 작성해 주었다.