[programmers] 같은 숫자는 싫어

 

큐/스택 이래서 큐랑 스택으로 풀었으나

public static int[] solution(int[] arr) {
    Stack<Integer> stack = new Stack<>();
    Queue<Integer> queue = new LinkedList<>();
    for (int i = 0; i < arr.length; i++) {
        if (i == 0) {
            stack.push(arr[i]);
            queue.add(arr[i]);
        }
        if (i > 0 && stack.peek() != arr[i]) {
            stack.push(arr[i]);
            queue.add(arr[i]);
        }
    }
    int k = 0;
    int[] answer = new int[queue.size()];
    while (!queue.isEmpty()) {
        answer[k] = queue.poll();
        k++;
    }
    return answer;
}

 

아래코드가 더 간결하네

list.getLast() 가 있었네 

public class Solution {
    public int[] solution(int []arr) {
        LinkedList<Integer> list = new LinkedList<Integer>();
        list.add(arr[0]);
        for(int i=1; i<arr.length;i++){
            if(arr[i]!=list.getLast()){
                list.add(arr[i]);
            }

        }
        Integer[] listing = list.toArray(new Integer[list.size()]);
        int []answer =Arrays.stream(listing).mapToInt(Integer::intValue).toArray();
        return answer;
    }
}