프로그래머스 lv2. 짝지어 제거하기
문제 설명
짝지어 제거하기는, 알파벳 소문자로 이루어진 문자열을 가지고 시작합니다.
먼저 문자열에서 같은 알파벳이 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을 반환합니다.
나의 코드
function solution(s) {
if(s.length % 2 !== 0) return 0
const arr = s.split("")
for(let i = 0; i < arr.length; i++) {
if(arr[i] === arr[i+1]) {
arr.splice(i, 2);
i = -1;
if(arr.length % 2 !== 0) return 0
}
}
return arr.length === 0 ? 1: 0;
}알고리즘은 맞지만, 시간 초과로 실패했다. 글자수에 따라 한 열 바퀴 스무 바퀴 아니 백 바퀴도 돌 수 있는 코드다. 비효율적이다.
효율적인 코드
function solution(s) {
let stack = []
for (let i = 0; i < s.length; i++) {
if (stack.length && stack[stack.length-1] === s[i]){
stack.pop()
}else{
stack.push(s[i])
}
}
return stack.length === 0 ? 1 : 0;
}이 코드의 원리는 문자열을 반복할 때 stack data structure를 사용하여 문자열의 문자를 추적하는 것이다.
stack을 사용하면 동일한 문자 쌍을 쉽게 확인하고, 발견하는 순간 제거할 수 있다.
이 코드는 for 루프를 사용하여 한 번에 한 문자씩 문자열을 반복한다.
각 문자에 대해 '현재 문자가 stack의 마지막 문자와 동일한지' 확인한다.
맞다면 stack에서 마지막 문자를 pop하여 동일한 문자 쌍을 제거한다. 그렇지 않은 경우, 현재 문자를 stack으로 push한다.
이러한 방식으로 문자열을 반복함으로써 코드는 전체 문자열을 통과하고,
모든 인접 문자 쌍을 검사하기 때문에 모든 경우를 검사할 수 있다.
반복이 끝날 때 stack이 비어 있으면 동일한 문자 쌍을 모두 제거했음을 의미하므로 함수는 1을 반환하고 그렇지 않으면 0을 반환한다.
이 솔루션의 장점은 아직 쌍을 이루지 못한 문자열만 추적하므로 동일한 문자 쌍을 발견하는 즉시 제거할 수 있다는 것이다.
그리고 제일 중요한 건, '단 한 바퀴'면 끝난다.
'동일한 문자 쌍'에 대한 문제가 출제된다면 stack에 pop, push하는 방식을 하나의 스킬로 사용해보자.