프로그래머스 181926번 수 조작하기 1 - Python, Java
문제 설명
정수 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 함수를 완성해 주세요.
제한 사항
- -100,000 ≤
n≤ 100,000 - 1 ≤
control의 길이 ≤ 100,000control은 알파벳 소문자 “w”, “a”, “s”, “d”로 이루어진 문자열입니다.
입출력 예시
예시 1
입출력 1
| n | control | result |
|---|---|---|
| 0 | “wsdawsdassw” | -1 |
입출력 예 설명 1
- 수
n은control에 따라 다음과 같은 순서로 변하게 됩니다. - 0 → 1 → 0 → 10 → 0 → 1 → 0 → 10 → 0 → -1 → -2 → -1
- 따라서 -1을 return 합니다.
코드 구현
Python
def solution(n, control):
for command in control:
if command == "w":
n += 1
elif command == "s":
n -= 1
elif command == "d":
n += 10
elif command == "a":
n -= 10
return n
Java
class Solution {
public int solution(int n, String control) {
int answer = n;
for (char c : control.toCharArray()) {
if (c == 'w') {
answer += 1;
} else if (c == 's') {
answer -= 1;
} else if (c == 'd') {
answer += 10;
} else if (c == 'a') {
answer -= 10;
}
}
return answer;
}
}