프로그래머스 LV0 수 조작하기 2

프로그래머스

 

문제

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

프로그래머스

 

풀이

정수 배열이 주어지게 해당 정수배열과 조건에 맞는 배열을 만들어 반환하는 문제이다. 딱히 어려움 없이 조건에 맞게 구현만 하면 되는 문제이다.

 

def solution(numLog):
    
    answer = []
    start = numLog[0]
    
    for i in numLog[1:]:
        
        if start + 1 == i:
            answer.append("w")
        elif start - 1 == i:
            answer.append('s')
        elif start + 10 == i:
            answer.append('d')
        elif start - 10 == i:
            answer.append('a')
        
        start = i
    
    return "".join(answer)

 

만약 dict와 zip을 이용해서 문제를 풀이한다면 좀 더 간결하게 코드를 작성할 수 있다. 

def solution(log):
    res=''
    joystick=dict(zip([1,-1,10,-10],['w','s','d','a']))
    for i in range(1,len(log)):
        res+=joystick[log[i]-log[i-1]]
    return res