프로그래머스 - 짝지어 제거하기(Swift)

프로그래머스 - 짝지어 제거하기(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/12973

프로그래머스

 

 

 

 나의 풀이

 

처음에는 문제에서 입력으로 주어지는 문자열을 스택으로 만들고 인덱스를 이용해서 지워나가려고 했는데 복잡할 것 같아서 문제 그대로 순차적으로 제거해나갔다.

하지만 문자열 길이 조건이 1,000,000이기 떄문에 배열을 여러번 탐색하는 것은 여러가지 케이스에서 시간초과가 났다.

 

그래서 빈 스택을 만들어서 한 번만 돌면 문제를 해결할 수 있도록 풀이했다.

 

import Foundation

func solution(_ s:String) -> Int{
    var stack: [Character] = []
    var arr = Array(s)
    
    for i in arr.indices {
        if stack.isEmpty { stack.append(arr[i])}
        else {
            if stack.last! == arr[i] {
                stack.removeLast()
            } else {
                stack.append(arr[i])
            }
        }
    }

    return stack.isEmpty ? 1 : 0
}

 

 

 

 피드백

 

배열을 사용할 때 last값과 removeLast함수를 적극적으로 이용하자.