[0단계] 가위 바위 보

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

프로그래머스

문제 설명

가위는 2 바위는 0 보는 5로 표현합니다. 가위 바위 보를 내는 순서대로 나타낸 문자열 rsp가 매개변수로 주어질 때, rsp에 저장된 가위 바위 보를 모두 이기는 경우를 순서대로 나타낸 문자열을 return하도록 solution 함수를 완성해보세요.

 

 

입출력 예

rsp	result
"2"	"0"
"205"	"052"

 

 

문제 풀이

보통이라면 if문을 써서

def solution(rsp):
    result = ''
    for i in rsp:
        if i == '2': result += '0'
        elif i == '0': result += '5'
        else: result += '2'
    return result

이렇게 작성하겠지만

 

 

def solution(rsp):
    case = {'2':'0', '0':'5', '5':'2'}
    return ''.join([case[x] for x in rsp])

이런 식으로 if 문없이 작성할 수 있다.

 

 

(심심해서 짜본 코드)

def solution(rsp):
    return ''.join(['5', '2', '0'][int(x)%4] for x in rsp)