[JAVA] 프로그래머스 Lv.1 : 같은 숫자는 싫어

 

예제:

 

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

 

제한 조건:

• 배열 arr의 크기 : 1,000,000 이하의 자연수
• 배열 arr의 원소의 크기 : 0보다 크거나 같고 9보다 작거나 같은 정수

---

Solution #1

import java.util.*;

public class Solution {
    public int[] solution(int[] arr) {
        Stack<Integer> stack = new Stack<>();
        for (int i : arr) {
            if (stack.isEmpty()) {
                stack.push(i);
            } else if (stack.peek() != i) {
                stack.push(i);
            }
        }

        int[] answer = new int[stack.size()];

        for (int i = stack.size() - 1; i >= 0; i--) {
            answer[i] = stack.pop();
        }

        return answer;
    }
}

 

Result #1

💡 Stack을 이용하여 풀었지만 효율성은 글쎄..?

Stack 자료형을 이용해서 arr의 원소를 차례대로 push() 한다. 여기서 조건이 있다.

1. Stack이 비어있을 경우, push() -> 첫 번째일 경우

2. Stack이 비어있지 않을 경우, peek()의 값과 arr의 원소 값이 다를 경우 push()

push()가 완료되면 이제 Stack의 값들을 역순으로 answer 배열에 저장한다.

---

Solution #2

import java.util.*;

public class Solution {
    public int[] solution(int[] arr) {
        List<Integer> arrayList = new ArrayList<>();
        
        int lastNum = -1;
        
        for(int i : arr){
            if(lastNum != i) arrayList.add(i);
            lastNum = i;
        }
        
        int[] answer = new int[arrayList.size()];
        for (int i = 0; i < arrayList.size(); i++) {
            answer[i] = arrayList.get(i);
        }

        return answer;
    }
}

 

Result #2

💡 ArrayList 사용!

최근 숫자를 저장하는 lastNum 변수를 생성하고, arr 배열의 원소 값과 비교해서 ArrayList에 add() 한다.arr 배열의 모든 원소 비교가 완료되면 ArrayList의 값들을 answer  배열에 저장한다.

 

---

 

More Algorithm!

 

👇👇

 

github.com/ggujangi/ggu.programmers

ggujangi/ggu.programmers

 

 

 

출처 : 프로그래머스