[프로그래머스/Java] Lv.0 무작위로 K개의 수 뽑기

문제

랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.

이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.

정수 배열 arr가 주어집니다. 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.

단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

- 제한사항

• 1 ≤ arr의 길이 ≤ 100,000
• 0 ≤ arr의 원소 ≤ 100,000
• 1 ≤ k ≤ 1,000

풀이

- 오답

요상한 결과가 나왔다. Arrays.fill()을 활용해 -1로 모든 값을 초기화 하고 시작했는데, -1은 어디로 간 것이며...
하지만 에러가 나는 것보다는 희망적이기 때문에 코드를 수정해 보기로 했다.

List 객체를 어차피 사용할 거라면, 주어지는 arr에서 중복값을 거른 다음 담아보기로 했다.

import java.util.Arrays;
import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int[] arr, int k) {
        int[] answer = new int[k];
        Arrays.fill(answer, -1);
        
        List<Integer> ansList = new ArrayList<>();
        for(int i=0;i<arr.length;i++){
            
            for(int j=0;j<ansList.size();j++){
                if(ansList.get(j) == arr[i]) break;
            }
            
            ansList.add(arr[i]);
        }
        
        for(int i=0;i<answer.length;i++){
            answer[i] = ansList.get(i);
        }
        
        return answer;
    }
}

- 정답

1. 주어지는 정수형 배열 arr을 순회하며 List 객체에 원소를 담는다. 이때, contains()로 List 객체의 arr[i] 포함 여부를 체크해 포함하지 않는다면 add() 한다.

2. 그리고 테스트 1번과 마찬가지로, 중복 여부와 관계 없이 ansList의 크기가 k와 같아졌다면 break;를 통해 반복문을 빠져나온다.
   ex. arr : [0, 1, 1, 2, 2, 3] - k : 3 → result : [0, 1, 2]

3. 반환 타입에 맞게 길이가 k인 배열을 선언한다.

4. 루프를 돌며 ansList.size()보다 작은 경우에는 ansList의 값을 get()으로 꺼내 배열에 저장한다. List 객체의 크기가 k보다 작은 경우 IndexOutOfBoundsException이 발생하니까 인덱스 값을 어떻게 설정해야 하는지 고민이었는데, 이렇게 경우의 수를 나누어 주면 되는 거였다.

5. i가 ansList의 값을 넘어서게 되면 answer[i] = -1;을 넣어준다.

import java.util.ArrayList;
import java.util.List;

class Solution {
    public int[] solution(int[] arr, int k) {
        List<Integer> ansList = new ArrayList<>();
        
        for (int i=0;i<arr.length;i++) {
            if(!ansList.contains(arr[i])) ansList.add(arr[i]);
            if(ansList.size() == k) break;
        }
        
        int[] answer = new int[k];
        for (int i=0;i<k;i++) {
            if(i < ansList.size()) {
                answer[i] = ansList.get(i);
            }else {
                answer[i] = -1;
            }
        }
        
        return answer;
    }
}

결과