[프로그래머스] 짝지어 제거하기

프로그래머스의 Lv.2 짝지어 제거하기 문제 풀이

문제 설명

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

예를 들어, 문자열 S = baabaa 라면
b aa baa → bb aa → aa →
의 순서로 문자열을 모두 제거할 수 있으므로 1을 반환합니다.

제한 사항

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

문제 풀이
짝이 지어진다면 제거를 하라는 문제를 읽고, 제거를 하려면 스택을 써야하나라는 생각이 들었다. 그래서 스택을 사용하면 문제를 풀었다. 먼저 스택이 비어있지 않거나 스택의 맨위의 글자와 c가 같은지 확인하고 같으면 제거하고, 아니면 푸쉬한다. 이를 반복하여 마지막에 스택이 비어있다면 1을 반환하고 아니면 0을 반환한다.

코드

public int solution(String s) {
   Stack<Character> stack = new Stack<>();

   for (char c : s.toCharArray()) {
     if (!stack.isEmpty() && stack.peek() == c) {
       stack.pop();
     } else {
       stack.push(c);
     }
   }
   if (stack.isEmpty()) {
     return 1;
   } else return 0;
 }