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

박채은·2023년 5월 7일
0

코딩테스트

목록 보기
33/52

문제

문제 풀이

스택을 사용해서 풀어야겠다고 생각했다.

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        List<Integer> stack = new ArrayList<>();
        stack.add(arr[0]);

        for(int i=0;i<arr.length;i++){
            if(stack.get(stack.size()-1) != arr[i]){
                stack.add(arr[i]);
            }
        }

        int[] answer = new int[stack.size()];
        int index = 0;
        for(Integer i: stack){
            answer[index++] = i;
        }
        return answer;
    }
}

0개의 댓글