프로그래머스 181925번 수 조작하기 2 - Python, Java
문제 설명
정수 배열 numLog가 주어집니다. 처음에 numLog[0]에서 부터 시작해 “w”, “a”, “s”, “d”로 이루어진 문자열을 입력으로 받아 순서대로 다음과 같은 조작을 했다고 합시다.
- “w” : 수에 1을 더한다.
- ”s” : 수에 1을 뺀다.
- “d” : 수에 10을 더한다.
- “a” : 수에 10을 뺀다.
그리고 매번 조작을 할 때마다 결괏값을 기록한 정수 배열이 numLog입니다. 즉, numLog[i]는 numLog[0]로부터 총 i번의 조작을 가한 결과가 저장되어 있습니다.
주어진 정수 배열 numLog에 대해 조작을 위해 입력받은 문자열을 return 하는 solution 함수를 완성해 주세요.
제한 사항
- 2 ≤
numLog의 길이 ≤ 100,000- -100,000 ≤
numLog[0]≤ 100,000 - 1 ≤
i≤numLog의 길이인 모든i에 대해|numLog[i] - numLog[i - 1]|의 값은 1 또는 10입니다.
- -100,000 ≤
입출력 예시
예시 1
입출력 1
| numLog | result |
|---|---|
| [0, 1, 0, 10, 0, 1, 0, 10, 0, -1, -2, -1] | “wsdawsdassw” |
입출력 예 설명 1
result인 “wsdawsdassw”를 따라 numLog[0]에서부터 시작해 조작을 하면 numLog의 값과 순서대로 일치합니다. 따라서 “wsdawsdassw”를 return 합니다.
코드 구현
Python
def solution(numLog):
answer = ''
for index in range(1, len(numLog)):
log = numLog[index] - numLog[index - 1]
if log == 1:
answer += "w"
elif log == -1:
answer += "s"
elif log == 10:
answer += "d"
elif log == -10:
answer += "a"
return answer
Java
class Solution {
public String solution(int[] numLog) {
StringBuilder answer = new StringBuilder();
for (int i = 1; i < numLog.length; i += 1) {
int num = numLog[i] - numLog[i - 1];
if (num == 1) {
answer.append("w");
} else if (num == -1) {
answer.append("s");
} else if (num == 10) {
answer.append("d");
} else if (num == -10) {
answer.append("a");
}
}
return answer.toString();
}
}