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

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

프로그래머스

 

접근 방식) Stack과 Queue 각각 사용해서 풀이

 

• Stack 사용

1) 배열의 0번째 원소는 미리 stack에 넣고,

    arr[i]와 arr[i - 1] 원소를 비교하며 중복되지 않으면 stack에 push 한다.

2) stack 원소들을 pop해서 후입선출이므로,  answer 배열 뒤에서부터 넣어준다.

import java.util.*;
public class Solution {
    public int[] solution(int[] arr) {
        Stack<Integer> stack = new Stack<>();
        stack.push(arr[0]);
        for (int i = 1; i < arr.length; i++) {
            if (arr[i] != arr[i - 1])
                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;
    }
}

 

• Queue 사용

1) 배열의 0번째 원소는 미리 queue에 넣고,

    arr[i]와 arr[i - 1] 원소를 비교하며 중복되지 않으면 queue에 add 한다.

2) queue 원소들을 poll해서 선입선출이므로,  answer 배열 앞에서부터 넣어준다.

import java.util.*;
public class Solution {
    public int[] solution(int[] arr) {
        Queue<Integer> queue = new LinkedList<>();
        queue.add(arr[0]);
        for (int i = 1; i < arr.length; i++) {
            if (arr[i] != arr[i - 1])
                queue.add(arr[i]);
        }
        
        int[] answer = new int[queue.size()];
        for (int i = 0; i < answer.length; i++) {
            answer[i] = queue.poll();
        }
        
        return answer;
    }
}