알고리즘/프로그래머스

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

반응형

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

 

코딩테스트 연습 - 짝지어 제거하기

짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾습니다. 그다음, 그 둘을 제거한 뒤, 앞뒤로 문자열을 이어 붙

programmers.co.kr


문제

문제 설명

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

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

b aa baa → bb aa → aa 

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

제한사항

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

입출력 예

s result
baabaa 1
cdcd 0

입출력 예 설명

입출력 예 #1
위의 예시와 같습니다.
입출력 예 #2
문자열이 남아있지만 짝지어 제거할 수 있는 문자열이 더 이상 존재하지 않기 때문에 0을 반환합니다.


풀이

풀이 과정

입력받은 문자열을 한글자씩 쪼개서 앞글자부터 비교하며 짝이 맞는 글자가 들어온다면 두 문자를 제거한다
이러한 과정을 반복하여 모든 문자를 제거할 수 있다면 1을 반환한다

앞글자 부터 하나씩 쌓아가며 비교하기 위해 스택을 사용

Stack<Character> stack = new Stack<>();

반복문을 이용하여 입력받은 문자열 s를 charAt으로 하나씩 쪼개 앞글자 부터 두 가지 경우로 나누어 처리한다

  1. 스택이 비어 있는 경우
    스택이 비어 있는 경우에는 문자를 그대로 추가한다

  2. 스택에 이미 문자가 들어있는 경우
    이미 문자가 있는 경우에는 또 다시 두 가지로 나누어 비교를한다
    스택에 있는 문자중 가장 마지막 문자가 현재 비교할 문자와 같다면 마지막 문자를 제거하고
    다르다면 다시 그대로 현재 문자를 추가한다
for(int i = 0; i < s.length(); i++) {
	char c = s.charAt(i);
	//스택이 비어있다면 다음 문자를 추가
	if(stack.empty()) {
		stack.push(c);
	}
	//스택에 문자가 들어있을 때
	else {
		//이전 문자와 현재 문자가 같다면 이전 문자를 제거
		if(stack.peek() == c) {
			stack.pop();
		}
		//다르다면 현재 문자를 추가
		else {
			stack.push(c);
		}
	}
}

 

반복이 끝난 후 스택에 문자가 남아있지 않다면 모든 문자열을 제거 할수 있으므로 1을 반환하고 그렇지 않다면 0을 반환한다

//스택에 문자가 남아 있지 않다면 성공
int answer = (stack.size() == 0)? 1 : 0;

 

최종 코드

import java.util.*;

class Solution
{
    public int solution(String s)
    {
        Stack<Character> stack = new Stack<>();
		for(int i = 0; i < s.length(); i++) {
			char c = s.charAt(i);
			//스택이 비어있다면 다음 문자를 추가
			if(stack.empty()) {
				stack.push(c);
			}
			//스택에 문자가 들어있을 때
			else {
				//이전 문자와 현재 문자가 같다면 이전 문자를 제거
				if(stack.peek() == c) {
					stack.pop();
				}
				//다르다면 현재 문자를 추가
				else {
					stack.push(c);
				}
			}
		}
		//스택에 문자가 남아 있지 않다면 성공
		int answer = (stack.size() == 0)? 1 : 0;

        return answer;
    }
}