[프로그래머스] Swift - 짝지어 제거하기 (LV.2)

👆문제풀러 가기👆

문제 설명

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙입니다. 이 과정을 반복해서 문자열을 모두 제거한다면 짝지어 제거하기가 종료됩니다. 문자열 S가 주어졌을 때, 짝지어 제거하기를 성공적으로 수행할 수 있는지 반환하는 함수를 완성해 주세요. 성공적으로 수행할 수 있으면 1을, 아닐 경우 0을 리턴해주면 됩니다.

 

예를 들어, 문자열 S = baabaa 라면

b aa baa → bb aa → aa →

의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

 

제한사항

• 문자열의 길이 : 1,000,000이하의 자연수
• 문자열은 모두 소문자로 이루어져 있습니다.

 

입출력 예제

s	result
"baabaa"	1
"cdcd"	0

 

풀이 [ 메모리: 21mb, 최대시간:  227.06ms ] 

func solution(_ s:String) -> Int{
    var stack = [Character]()
    
    for char in s {
        if let before = stack.last {
            if before == char {
                stack.removeLast()
                continue
            }
        }
        stack.append(char)
    }
    
    return stack.isEmpty ? 1 : 0
}

문제를 받고 생각한 것.

1. 문자열 내에 같은 문자가 연달아 있는지 확인해야 한다.
2. 같은 문자가 연달아 오는지 확인하는 방법으로는, 반복문의 인덱스를 활용해 앞뒤를 비교하거나 stack을 활용해 새로운 배열을 만드는 방법이 있다.
3. 반복문의 인덱스를 활용해 해당 배열에서 제거하면 index Error를 경험하거나 풀이가 까다로워질 수 있다. 그러니 stack을 활용해 새로운 문자 배열을 만든다.
4. stack에서 맨 뒤의 문자 2개를 비교하는 방법도 있지만, stack에 문자(char)를 담기 전에 stack에 담길 문자(char)가 stack의 맨 뒤 문자(stack.last)와 같은지 미리 확인할 수 있다.
5. char와 stack.last가 같다면 char를 stack에 담지 않고 stack.last를 제거(stack.removeLast())한다. ( = stack에 담은 뒤에 2개 제거한 것과 같음 )
   baabaa = b -> ba -> b -> [] -> a -> [] ( char를 담기 전 제거 )
   baabaa = b -> ba -> baa -> b -> bb -> [] -> a -> aa -> [] ( char를 담은 뒤 제거 )
6. 만약 다르다면 char를 stack에 담는다.
7. 위의 과정을 반복하여 stack이 비어있다면 1, 비어있지 않다면 0을 return한다.