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

1. Input, Output, Example

랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 한다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 한다. 이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해보자.

- 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 반환

 

2. Constraint

1) 1 ≤ arr의 길이 ≤ 100,000

2) 0 ≤ arr의 원소 ≤ 100,000

3) 1 ≤ k ≤ 1,000

 

3. Code

1) 첫 코드(2023/04/27)

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int k) {
        int[] answer = new int[k];
        ArrayList<Integer> list = new ArrayList<>();
        int i = 0;
        while(list.size()<k && i<arr.length){
            if(!list.contains(arr[i]))
               list.add(arr[i]);
            i++;
        }
        for(i=0 ; i<list.size() ; i++)
            answer[i] = list.get(i);
        if(list.size()!=k)
            for(i=list.size() ; i<k ; i++)
                answer[i] = -1;
        return answer;
    }
}