본문 바로가기

공부/알고리즘

[Java] 코딩테스트 연습_3

문제 설명

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

출처


 

두 가지로 만들어 보았다.

하나는 Array만 쓰는 거고, 하나는 ArrayList 쓰는 거임.

 

Array

public class Solution {
    public int[] solution(int []arr) {
        int newLength = 0, newIndex = 1;
        
        for(int i=1; i<arr.length; i++){
            if(arr[i-1] != arr[i]){
                newLength++;
            }
        }
        
        int[] answer = new int[newLength+1];
        answer[0] = arr[0];
        
        for(int j=1; j<arr.length; j++){
            if(arr[j-1] != arr[j]){
                answer[newIndex] = arr[j];
                newIndex++;
            }
        }
        
        return answer;
     }
 }

많은 방식이 있겠지만,

나는 첫번째 인덱스(0번째)의 값은 고정한 뒤,

arr의 인덱스를 1부터 시작해서 -1번째 값과 비교하여 

다른 값을(중복되지 않는 값을) answer array에 넣어주는 식으로 소스를 짜보았다.

(사실 인덱스를 0부터 시작하게 하니까 마지막 값을 비교하기가 까다로워서...

마지막 횟수를 돌 때 if를 한 번 더 써주어야 해서 번거로워서 인터넷 찾아보다가 걍 바꿈)

 

방식은 똑같고 ArrayList를 썼을 때

import java.util.*;

public class Solution {
    public int[] solution(int []arr) {
        ArrayList<Integer> arrayList = new ArrayList<>();
        
        arrayList.add(arr[0]);
        for(int i=1; i<arr.length; i++){
            if(arr[i-1] != arr[i]){
                arrayList.add(arr[i]);
            }
        }
        
        int[] answer = new int[arrayList.size()];
        for(int j=0; j<arrayList.size(); j++){
            answer[j] = arrayList.get(j);
        }
        
        return answer;        
    }
}

새로운 array를 위한 length를 따로 지정하지 않아도 되어서 좀 더 수월하다고 느낌.

 

 

 

 

미노랑 공부하다가 풀게된 알고리즘 문제인데,

봤을 때에는 쉬워보이는데 막상 만드려고 하니 Array라서 좀 번거로왔다,,,

 

암튼 잘 풀었음!!!

 

반응형