728x90
문제 설명
배열 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));
}
}728x90
'Java 알고리즘 공부 (프로그래머스)' 카테고리의 다른 글
| [프로그래머스 Lv1] 문자열 내 마음대로 정렬하기 (Java) (0) | 2023.09.16 |
|---|---|
| [프로그래머스 Lv1] 나누어 떨어지는 숫자 배열 (Java) (1) | 2023.09.16 |
| [프로그래머스 Lv 1] 가운데 글자 가져오기 (0) | 2023.09.16 |
| [프로그래머스 Lv1] 2016년 (Java) (0) | 2023.09.16 |
| [프로그래머스 Lv1] 폰켓몬 (Java) (0) | 2023.09.16 |