프로그래머스 - 같은 숫자는 싫어

⭐ Lv. 0 - 같은 숫자는 싫어

✅ 문제 설명

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

• arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
• arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

✅ 제한사항

• 배열 arr의 크기 : 1,000,000 이하의 자연수
• 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

✅ 입출력 예

arr	answer
[1,1,3,3,0,1,1]	[1,3,0,1]
[4,4,4,3,3]	[4,3]

✅ 풀이 코드 + 설명

단순한 stack의 개념을 활용한 문제입니다. 배열의 길이가 0일 경우와 0이 아닐 경우로 구분하여 아래와 같이 코드를 작성하였습니다.

function solution(arr)
{
    if(arr.length === 0) return arr;
    else {
        let answer = [];
        answer.push(arr[0]);
    
        for(num of arr) {
            if(num !== answer[answer.length - 1]) answer.push(num);
        }
    
        return answer;
    }
}

위 코드에서 다음과 같이 더 개선할 수 있습니다.

• 새로운 answer 배열을 생성함과 동시에 arr의 첫 원소를 담기
• for문을 index를 기준으로 탐색하여 길이가 0이더라도 정상적으로 빈 배열이 return되도록 하기

function solution(arr)
{
    let answer = [arr[0]];
    
    for(let i = 1; i < arr.length; i++) {
        if(arr[i] !== answer[answer.length - 1]) answer.push(arr[i]);
    }
    
    return answer;
}

이보다 더 간결한 답변도 존재했습니다. filter를 이용하면 아래와 같이 더 간결한 코드로 완성할 수 있습니다.

const solution = (arr) => arr.filter((cur, index, array) => cur !== array[index + 1]);

현재 배열 요소와 다음 배열 요소를 비교하여 서로 같지 않은 경우만 필터링합니다. 즉, 연속해서 중복되는 값들 중 가장 마지막 값만 모아 return하게 됩니다. 가장 마지막의 경우는 undefined와 비교하게 되므로 정상적으로 값이 담겨 return됩니다.