[코딩 테스트 연습 입문] 컨트롤 제트

문제

숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.

 

❗제한사항❗

• 1 ≤ s의 길이 ≤ 200
• -1,000 < s의 원소 중 숫자 < 1,000
• s는 숫자, "Z", 공백으로 이루어져 있습니다.
• s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
• 연속된 공백은 주어지지 않습니다.
• 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
• s는 "Z"로 시작하지 않습니다.
• s의 시작과 끝에는 공백이 없습니다.
• "Z"가 연속해서 나오는 경우는 없습니다.

 

입출력 예

 

입출력 예 설명

입출력 예 #1

• 본문과 동일합니다.

입출력 예 #2

• 10 + 20 + 30 + 40 = 100을 return 합니다.

입출력 예 #3

• "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.

입출력 예 #4, #5

설명 생략

 

✔풀이

def solution(s):
    answer = 0

    s=s.split(' ')

    for i in range(len(s)):
        if s[i]=='Z':
            answer -= int(s[i-1])
        else:
            answer+=int(s[i])

    return answer

📝설명

문자열은 공백으로 숫자와 Z 사이를 구분짓고 있다.

그래서 문자열을 리스트로 바꾸면서, 공백을 기준으로 문자열을 분할하고 싶다는 생각을 했다.

구글링해보니 .split('구분짓는 기준')을 이용하면 원하는 기준으로 문자열을 리스트로 바꿀 수 있다고 한다.

따라서 문자열 s를 s.split(' ')를 통해 공백을 기준으로 한 리스트를 만들면서 s를 재정의하였다.

이렇게 되면 입출력 예시 1번의 s는 ["1", "2", "Z", "3"]로 재정의된다!

Z의 경우, Z 바로 전에 더했던 숫자를 뺀다고 했다.

제한사항을 보면 Z가 연속해서 나오는 경우는 없다고 했기 때문에, Z 바로 앞에 있는 것을 뺀다는 것과 같은 말이다.

 인덱스 넘버를 사용할 생각이어서, for문에 range(len(s))를 반복범위로 넣었다.

만약 s[i]가 Z라면, 그 전의 인덱스 넘버를 빼야 한다.

따라서, answer-=int(s[i-1])을 넣었다.

int를 하는 이유는, 리스트의 각 요소의 type은 현재 문자열이기 때문이다.

s[i]가 Z가 아닌 경우에는 (아까 공백을 기준으로 리스트를 만들면서 공백을 없앴기 때문에) 숫자밖에 없다.

따라서, answer+=int(s[i])를 쓴다.

 

👑모범 답안

 

👀느낀 점

: 조금 길긴 했지만, 어쨌든 모범 답안과 유사한 형태!!

나는 뿌듯합니다...

잊지말자 .split()