[GO] 프로그래머스 > 코딩 테스트 입문 > LV.0 컨트롤 제트 문제

문제 설명 

- 숫자와 "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. Z 가 나온 다음에 그 이전값을 다시 빼주어야 한다는 부분에서 Stack 을 활용해서 마지막에 사용한 값을 다시 불러오고 빼는게 베스트라고 생각 했습니다. 

 

2. 그래서 아래와 같이 임의로 Stack 을 만들어서 사용 했습니다. 

 

3. 이후 string 을 split 을 활용해서 space 를 기준으로 분리 

 

4. for 문을 활용해서 값을 확인

 

5. Z 가 아닐 경우 값을 Stack 에 값을 추가 및 sum 에 추가 

 

6. Z 일 경우 값을 pop 하고 sum 에서 마이너스 함. 

 

7. 계산이 완료 된 이후 다시 return 해줌.

 

import (
    "strings"
    "strconv"
)

type Stack struct {
    items []int
}


func(s *Stack) Push( data int) {
    s.items = append(s.items, data)
}

func (s *Stack) Pop() int {
    temp := s.items[len(s.items)-1]
    s.items = s.items[:len(s.items) -1 ]
    return temp
}

func solution(s string) int {
    
    splitStr := strings.Split(s, " ")
    stack := new(Stack)
    sum := 0
    
    for _ , v := range splitStr {
        if v != "Z"{
            resultInt, _ := strconv.Atoi(string(v))
            stack.Push( resultInt)
            sum += resultInt
        } else {
            minus:= stack.Pop()
            sum -= minus 
        }
    }
    return sum
}

 

개선 사항? 

 

1. stack 에 isEmpty 함수 추가?