[programmers] 짝지어 제거하기 - Java
"짝지어 제거하기"는 문자열에서 같은 알파벳이 2개 붙어 있는 짝을 찾아 제거하는 과정을 반복하여 문자열을 모두 제거하는 문제입니다. 처음에는 ArrayList를 사용하여 문제를 해결하려 했지만 효율성 테스트를 통과하지 못했습니다. 이후 스택 자료구조를 사용하여 코드를 개선하였고, 이를 통해 선형 시간만에 문자열을 검사할 수 있게 되었습니다.
Feb 25, 2024
짝지어 제거하기
문제 설명
짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다. 먼저 문자열에서 같은 알파벳이 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을 반환합니다.
처음 시도한 방식
solution.java
import java.util.*; class Solution { public int solution(String s) { List<String> list = new ArrayList<>(Arrays.asList(s.split(""))); for(int i=0; i<list.size()-1; i++){ String s1 = list.get(i); String s2 = list.get(i+1); if(s1.equals(s2)){ list.remove(i); list.remove(i); i=0; } } if(list.size()==2){ String s1 = list.get(0); String s2 = list.get(1); if(s1.equals(s2)){ return 1; } } return 0; } }
개선한 코드
solution.java
import java.util.*; class Solution { 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); } } return stack.isEmpty() ? 1 : 0; } }
핵심 키워드
- 리스트의 경우 요소를 제거할 때마다 해당 요소 뒤의 모든 요소가 이동해야 하므로 시간이 많이 소요된다. 특히 이 작업을 반복하면서 문자열이 매우 길어질수록 성능이 더욱 저하된다.
- 따라서 요소를 반복적으로 제거하는 코드의 경우 더 효율적인 접근 방식은 스택 자료구조를 사용하는 것이다. 스택을 사용했을 경우 중복된 문자를 찾으면 스택에서 가장 위에 있는 값만 제거해버리면 된다. 이를 통해 선형 시간만에 문자열을 검사할 수 있다.
결론!
처음 해당 문제를 접근했을 때 ArrayList로 코드를 작성했고, 효율성 테스트를 통과하지 못했다. 이를 스택 자료구조를 활용함으로써 코드를 개선시킬 수 있었다.
Share article