수 조작하기 1

송용진·2023년 8월 9일
0

정수 n과 문자열 control이 주어집니다.
control은 "w", "a", "s", "d"의 4개의 문자로 이루어져 있으며,
control의 앞에서부터 순서대로 문자에 따라
n의 값을 바꿉니다.

"w" : n이 1 커집니다.
"s" : n이 1 작아집니다.
"d" : n이 10 커집니다.
"a" : n이 10 작아집니다.
위 규칙에 따라 n을 바꿨을 때
가장 마지막에 나오는 n의 값을
return 하는 solution 함수를 완성해 주세요.

내 코드

def solution(n, control):
    answer = 0
    for i in range(len(control)):
        if control[i] == "w":
            n = n + 1
        elif control[i] == "s":
            n = n - 1
        elif control[i] == "d":
            n = n + 10
        elif control[i] == "a":
            n = n - 10
        else:
            continue
    answer = n
    return answer

예시 코드

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

0개의 댓글