💛 오늘의 학습 키워드

[스택/큐] 같은 숫자는 싫어

📝 문제

문제 설명

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

입출력 예

arr	answer
[1,1,3,3,0,1,1]	[1,3,0,1]
[4,4,4,3,3]	[4,3]

💬 내 풀이

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
            
        ArrayList<Integer> arrList = new ArrayList<>();
        
        arrList.add(arr[0]);
        for(int i = 1; i < arr.length; i++) {
            if(arr[i-1]!=arr[i]) {
                arrList.add(arr[i]);
            }
        }
        
        return arrList.stream().mapToInt(i->i).toArray();
    }
}

💻 내 접근 방법

1. ArrayList 객체를 생성한 후 index=0에 있는 값을 먼저 add()를 사용하여 추가한다.

2. for()문을 사용해서 arr[i-1]의 값이 arr[i]의 값과 같을 때만 ArrayList 객체에 값을 넣어준다.

3. stream의 mapToInt()를 사용해서 IntStream으로 변환 후 toArray()를 사용해서 int형 Array 객체로 변환하여 return 한다.

✨ 다른 사람 풀이

import java.util.*;

public class Solution {
    public Stack<Integer> solution(int []arr) {

        Stack<Integer> stack = new Stack<>();

        for(int num : arr){
            if(stack.size() == 0 || stack.peek() != num){
                stack.push(num);
            }
        }



        return stack;
    }
}

1. Stack 객체를 먼저 생성한 후 for()문을 사용한다.

2. 첫번째 값은 먼저 push()를 사용해서 스택에 추가하고, 이후에는 peek() 메소드를 사용해서 스택에 있는 값과 arr에 있는 값을 비교하여 같지 않을 때만 스택에 추가한다.

   Stack 특징

   • LIFO(Last In First Out) 구조를 가진다.
   • 단방향 입출력 구조를 가진다. (들어오는 방향과 나가는 방향이 같음)
   • 데이터를 하나씩만 넣고 뺄 수 있다.

   • push() : 데이터를 스택에 추가한다.
   • peek() : 스택의 마지막 요소를 반환하며, 스택 값은 변화를 주지 않는다.

3. return 타입을 변경해서 Stack 을 바로 리턴한다.

💦 회고

자바의 stack 클래스를 사용할 일이 많지 않아서 사용해야 겠다는 생각을 못했다. 그리고 주어진 리턴타입을 꼭 지킬 필요가 없다는 것을 처음 알았다..!

참고
https://ittrue.tistory.com/200