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

🖊️문제

 

배열 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함수를 완성해 주세요.

 

 

 

 

🖊️풀이

 public static int[] solution(int []arr) {
        
        //stack 생성해서 배열에 저장된 값 넣어주기
        //stack이 비어있거나 peek한 값과 같지 않다면 값 저장
        Stack<Integer> stack = new Stack<>();
        for(int i : arr){
            if(stack.isEmpty()){
                stack.push(i);
            }else if(stack.peek()!=i){
                stack.push(i);
            }
        }
        
        //stack에 저장했던 값 배열에 저장시켜 반환
        int[] answer = new int[stack.size()];
        for (int i = stack.size()-1; i >=0 ; i--) {
            answer[i] = stack.pop();

        }

        return answer;
    }