[99클럽] 코딩테스트 스터디, 11일차 TIL - 비기너 (프로그래머스, 같은 숫자는 싫어)

 

1. 문제풀이

1) 문제

배열 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보다 작거나 같은 정수

 

2) 해석

- 연속적으로 나타나는 숫자, 하나만 남기고 제거
- 순서 유지

 

 

3) 풀이

function solution(arr) {
    const answer = [];
    let prevNum;
    
    arr.forEach((num, i) => {
        if(i === 0 || prevNum !== num){
            answer.push(num);
            prevNum = num
        } 
    })
    return answer;
}

 

문제를 보자마자 for문으로 풀면 되겠다고 생각하면 forEach를 사용했다.

확실히 기본 for문보다 적어야 하는 코드 수가 줄어서 편하다.

 

우선 최종적으로 리턴할 answer 배열을 선언해주고, 이전 숫자를 담을 prevNum 변수를 만들어준다.

 

그 후, forEach를 통해 arr를 돌고, i가 0이거나 현재 숫자가 이전 값과 같지 않으면 answer에 값을 넣어주도록 만들었다.

그리고 prevNum을 새로운 값으로 없데이트 해줬다.  

 

 

- 걸린시간

 

2. 리팩토링

1) 다른 사람 풀이

function solution(arr)
{
    return arr.filter((val,index) => val != arr[index+1]);
}

 

어떤 식으로 리팩토링을 해야하지 고민하다가 다른 사람 코드를 봤더니 filter를 사용한 사람이 있었다.

내가 했던 풀이랑 시간복잡도는 같은데, 훨씬 깔끔했다.

 

filter를 사용할 생각은 못했는데!! 

역시 세상은 넓고 천재는 많다,,