정수 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])