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

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

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

 

문제 설명

• 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 모두 제거
• 연속된 숫자들 제거 후 반환될 때 arr의 원소 순서 유지

제한사항

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

 

내 풀이

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        int[] answer = {};
        ArrayList<Integer> list =new ArrayList<>();
        
        for(int i=1;i<arr.length;i++){
            if(arr[i]==arr[i-1])
                continue;
            else
                list.add(arr[i-1]);
        }
        
        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;
    }
}

우선 가변적인 크기를 가질 수 있는 ArrayList를 이용했다.

반복문을 통해 배열 arr를 i=1부터 현재 위치의 원소 값과 그 위치의 앞의 원소 값을 비교하며 검사하였고 같을 때는 패스하고 같지 않을 때 그 전의 원소 값을 list에 추가했다.

또한 index밖으로 밀려 검사가 불가능했던 마지막 원소는 list에 직접 추가했다.

 

 

다른 풀이

import java.util.*;

public class Solution {
    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;
    }
}

arr 원소 크기는 0~9이기 때문에 무의미한 값으로 preNum=10이라고 선언해준다.

반복문을 통해 arr의 원소들과 preNum값을 비교해 같지 않을 때 tempList에 원소값을 추가한다.

preNum 값을 arr의 원소값으로 변경한다.

 

 

내 풀이보다 더 깔끔하다..