프로그래머스의 Lv1 문제
같은 숫자는 싫어입니다.
문제를 먼저 살펴보겠습니다.
문제 설명
배열 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 함수를 완성해 주세요.입출력 예
예시1) arr [1,1,3,3,0,1,1] answer [1,3,0,1]
예시2) arr [4,4,4,3,3] answer [4,3]
예시 1번을 잘 보시면,
중복된 숫자를 모두 제거하는게 아니라
연속되는 숫자가 중복되지 않아야합니다.
중복된 숫자를 모두 제거하라고 했으면 Set을 이용하면 쉽게 풀 수 있겠죠?
저는 먼저 배열 arr의 첫번째 숫자를 first라는 변수에 담아주고
리스트를 하나 만들어서 첫번째 숫자 first를 리스트에 넣어 준 다음에
반복문을 돌려서 first와 배열의 다음 숫자를 비교 한 뒤,
first와 다음 숫자가 같지 않다면 list에 add해주도록 짜 보았습니다.
그 다음 리스트를 배열로 바꿔서 리턴해주면 간단할 것 같았습니다.
코드를 한번 볼까요?
public static int[] solution(int[] arr){
List<Integer> list = new LinkedList<>();
int first = arr[0];
list.add(first);
for (int a : arr){
if (first != a){
list.add(a);
first = a;
}
}
return list.stream().mapToInt(Integer::intValue).toArray();
결과를 한번 살펴보겠습니다.
제출도 해보겠습니다.!
네 생각한 것처럼 결과가 잘 나왔네요!
다른 분들은 어떻게 풀었을까요??
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;
}
제가 푼 것과 비슷하지만
저는 배열의 첫번째 숫자를 first에 넣어줬는데
이분은 임의의 수인 10을 넣어서 비교를 해주고 있네요.
임의의 수를 넣어서 하게 되면 first를 리스트에 추가하지 않아도 되니까
코드가 더 간결해질 것 같습니다!