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

Rae-eun Yang·2022년 9월 4일
2

프로그래머스

목록 보기
20/83

문제 설명


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


입출력 예

arranswer
[1,1,3,3,0,1,1][1,3,0,1]
[4,4,4,3,3][4,3]

입출력 예 설명

입출력 예 #1,2

문제의 예시와 같습니다.


풀이 코드 (Java)

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        
        Stack<Integer> s = new Stack<>();

        for(int i : arr) {
            
            int cur = i;
            
            if(!s.isEmpty() && s.peek() == cur) {
                continue;
            }
            
            s.push(cur);
            
        }
        
        int[] answer = new int[s.size()];
        int curIndex = s.size();
        while(!s.isEmpty()){
            answer[--curIndex] = s.pop();
        }
 
        return answer;
        
    }
}

스택으로 처리 가능한 간단한 문제


풀이 코드 (JavaScript)

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

자바스크립트로 스택을 구현하려고 여러 가지를 참고했었지만 (스택 클래스를 구현한다거나..) 간단하게 배열을 사용해서 풀 수 있었던 문제


profile
ContentsDev Manager

3개의 댓글

comment-user-thumbnail
2022년 9월 5일

좋은 풀이 감사합니다

1개의 답글
comment-user-thumbnail
2022년 9월 6일

if(arr[i] !== arr[i + 1]){
answer.push(arr[i]);
}
저의 경우 배열의 i번째와 i+1번째를 비교하여 풀었는데
배열의 길이가 0일때 answer에 먼저 요소를 push해주고 그 이후에 answer의 요소와 비교하는 방식도 너무 좋은것같습니다!
또 하나 배워갑니다 🤩

답글 달기