[프로그래머스 Lv1] 같은 숫자는 싫어 (Java)

문제 설명

배열 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]

입출력 예 설명

입출력 예 #1,2
문제의 예시와 같습니다.

---

코드

import java.util.ArrayList;

public class Solution26_1 {

	public static int[] solution(int[] arr) {
		int[] answer = {};

		ArrayList<Integer> list = new ArrayList<>();

		int idx = 0;

		for (int i = 1; i < arr.length; i++) {
			if (arr[idx] != arr[i]) {
				list.add(arr[idx]);
				idx = i;
			}
		}
		
		list.add(arr[idx]); // 마지막 바뀐 인덱스는 추가해줘야함. // list.add(arr[arr.length-1]); 로 써도 동일

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

		for (int i = 0; i < answer.length; i++) {
			answer[i] = list.get(i);
		}

		return answer;
	}

	public static void main(String[] args) {
		int[] arr = { 1, 1, 3, 3, 0, 1, 1 };
		int[] arr1 = { 4, 4, 4, 3, 3 };

		int re[] = solution(arr);
		System.out.println(Arrays.toString(re)); //Arrays.toString()은 배열 내용을 [1, 3, 0, 1] 과 같은 형태로 출력해준다.

		int re1[] = solution(arr1);
		System.out.println(Arrays.toString(re1));

	}

}

idx로 기준을 잡아주는 것이 중요하다.

 

다른 더 간단한 코드

public class Solution26_1 {

	public static int[] solution(int[] arr) {
		int[] answer = {};

		ArrayList<Integer> list = new ArrayList<>();

		int num = -1; // -1로 설정하고

		for (int i = 0; i < arr.length; i++) {// 0으로 출발해서 추가하는 기준을 바꾸면 된다.
			if (num != arr[i]) {// (아까와 다르게 이 코드는 다르면 arr[i]를 추가)
				list.add(arr[i]);
				num = arr[i];
			}
		}
        
        //마지막 인덱스 추가해주는 코드 사라짐

		answer = new int[list.size()]; // answer 배열 빈 공간이라서 할당해줘야 함.

		for (int i = 0; i < answer.length; i++) {
			answer[i] = list.get(i);
		}

		return answer;
	}

	public static void main(String[] args) {
		int[] arr = { 1, 1, 3, 3, 0, 1, 1 };
		int[] arr1 = { 4, 4, 4, 3, 3 };

		int re[] = solution(arr);
		System.out.println(Arrays.toString(re)); // Arrays.toString()은 배열 내용을 [1, 3, 0, 1] 과 같은 형태로 출력해준다.

		int re1[] = solution(arr1);
		System.out.println(Arrays.toString(re1));

	}

}