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

 

🍀 문제

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다. 단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다.

 

배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

 

입출력 예

[1,1,3,3,0,1,1]	[1,3,0,1]
[4,4,4,3,3]	[4,3]

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

코딩테스트 연습 - 같은 숫자는 싫어

 

 

나의 코드 😊

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        int lastNum = arr[0];// 이전 숫자 (첫 번째 요소로 시작)
		int cnt = 1;// 답변 배열의 크기

		for (int i = 1; i < arr.length; i++) {
			if (arr[i] == lastNum) {// 이전 숫자와 연속되면 0~9에 속하지 않는 -1로 대입
				arr[i] = -1;
			} else {// 이전 숫자와 연속되지 않을때
				lastNum = arr[i];// 이전 숫자 바꿈
				cnt++;// 배열의 크기를 늘림
			}
		}

		int[] answer = new int[cnt];
		int index = 0;
		for (int i = 0; i < arr.length; i++) {
			if (arr[i] != -1) {// -1로 대입한 요소가 아닌 숫자만 답변 배열에 넣기
				answer[index++] = arr[i];
			}
		}

		return answer;
    }
}

 

 

그리고 어제 배운 stream을 이용한 코드 👇

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        int lastNum = -1;
		ArrayList<Integer> list = new ArrayList<>();

		for (int i = 0; i < arr.length; i++) {
			if (arr[i] != lastNum) {
				list.add(arr[i]);
				lastNum = arr[i];
			}
		}

		int[] answer = list.stream().mapToInt(i -> i).toArray();
		return answer;
    }
}