#28😊 같은 숫자는 싫어 with Stack

[문제]

출처 : https://school.programmers.co.kr

---

[해답]

- 문제 조건 자체가 Stack 혹은 Queue를 이용해 풀라고 했기에 Stack을 이용해 풀었다.

import java.util.*;
import java.util.stream.*;

public class Solution {
    public int[] solution(int []arr) {
        Stack<Integer> stack = new Stack<>();
        
        // 제네릭에는 원시형 타입이 들어가지 않기에 Integer List로 변환
        List<Integer> arrInteger = Arrays.stream(arr).boxed().collect(Collectors.toList());
        
        for(Integer num : arrInteger) {
            if(stack.empty() || (stack.peek() != num)) { // stack이 비어 있을 때도 체크해줘야 함
                stack.push(num);
            }
        }
        
        // Stream을 이용해 Stack을 int Array로 변환
        int[] answer = stack.stream().mapToInt(Integer::intValue).toArray();

        return answer;
    }
}

 

- 통과는 했는데 아마 Stream을 2번 사용해서 효율은 그닥 좋지 않은 것 같다.

 

- 이렇게 더 간략(?)하게 풀 수도 있었다.(일주일 지났는데 우연히 생각나서 다시 한 번 풀어봄)

- 위 풀이와 비교했을 때 arrInteger List를 굳이 만들 필요가 없었다.

import java.util.*;
import java.util.stream.*;

public class Solution {
    public int[] solution(int []arr) {        
        Stack<Integer> stack = new Stack<>();

        for(int num : arr) {
            if(stack.empty() || !(stack.peek().intValue() == num)) { // Integer 주소 말고 숫자값 그 자체를 비교
                stack.push(Integer.valueOf(num));
            }
        }
        
        int[] answer = stack.stream().mapToInt(Integer::intValue).toArray();
        return answer;
    }
}

 

cf)

- 아래와 같은 식으로 반복문만을 이용해서 풀수도 있긴 한데... 문제 전제가 그것이 아니었기에 일부러 사용하지 않았다.

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        ArrayList<Integer> tempList = new ArrayList<Integer>();
        int preNum = 10;
        for(int num : arr) {
            if(preNum != num) {
                tempList.add(num);
            }
            preNum = num;
        }       
        int[] answer = new int[tempList.size()];
        for(int i=0; i<answer.length; i++) {
            answer[i] = tempList.get(i).intValue();
        }
        return answer;
    }
}