[프로그래머스] Java 코딩테스트 - 무작위로 K개의 수 뽑기

무작위로 K개의 수 뽑기

✅ 문제 설명

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

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

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

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

✅ 제한사항

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

🔽 소스코드 1

💻 풀이

• 주어진 배열 arr에서 중복되지 않는 숫자만 뽑아내기 위해 동적으로 크기가 조절되는 ArrayList를 사용한다.

• for 반복문을 사용하여 arr의 각 요소를 확인한다.
  → list.contains() 메서드로 현재 숫자가 이미 리스트에 포함되어 있는지 확인하고,
  → 포함되어 있지 않은 경우에만 list.add()로 숫자를 추가한다.
  → 리스트의 크기가 k에 도달하면 더 이상 숫자를 추가하지 않습니다.

• k 크기의 정수 배열을 생성한 후, 리스트에 담긴 숫자들을 순서대로 배열에 옮겨 담아준다.
  → 리스트에 담긴 숫자의 개수가 k보다 적다면, 남은 공간은 -1로 채워준다.

⌛ 시간 0.04ms ~ 10.51ms

	public int[] solution(int[] arr, int k) {
		
		List<Integer> list = new ArrayList<>();
		
		for(int num : arr) {
			if(list.size() < k && !list.contains(num)) {
				list.add(num);
			}
		}
		
		int[] arr_list = new int[k];
		
		for(int i = 0; i < k; i++) {
			if(i < list.size()) {
				arr_list[i] = list.get(i);
			}else {
				arr_list[i] = -1;
			}
		}
        
        return arr_list;
    }