프로그래머스

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

빙수빈수 2023. 10. 9. 11:04

https://school.programmers.co.kr/learn/courses/30/lessons/12906

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

[문제]

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

 

[코드]

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        Stack<Integer> stack=new Stack<>();
        
        for(int i=0;i<arr.length;i++){
            if(stack.isEmpty()||stack.peek()!=arr[i])
                stack.push(arr[i]);
        }
        
        
        int[] answer=new int[stack.size()];
        for(int i=answer.length-1;i>=0;i--)
            answer[i]=stack.pop();
        
        return answer;
    }
}

 

[고찰]

 이번 문제는 연속해서 나오는 수는 한 번만 출력해야 하므로 가장 나중에 삽입된 값을 가져올 수 있는 stack을 사용하여 해결하였다.