본문 바로가기
python 코딩 공부

프로그래머스 수 조작하기1

by Gustav7 2024. 10. 12.
def solution(n, control):
    for i in control:
        if i == 'w':
            n += 1
        elif i == 's':
            n -= 1
        elif i == 'd':
            n += 10
        elif i == 'a':
            n -= 10
       

 

다른 사람의 풀이중 인상 깊은 풀이 방법

 

def solution(n, control):
    key = dict(zip(['w','s','d','a'], [1,-1,10,-10]))
    return n + sum([key[c] for c in control])

 

zip 함수를 이용한 key값과 value 값의 매칭을 통한 sum함수로의 마무리!!

zip함수의 쓰임법도 중요시하자!!