[프로그래머스(Java)] 무작위로 K개의 수 뽑기 / ArrayList.contains(), Stream concat(), limit(), distinct()

 

 

Level. 0

 

문제

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

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

정수 배열 arr가 주어집니다.
문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.
단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

* 제한사항
- 1 ≤ arr의 길이 ≤ 100,000
- 0 ≤ arr의 원소 ≤ 100,000
- 1 ≤ k ≤ 1,000

 

풀이

import java.util.*;

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

 

ArrayList.contains(Object o)

• 리스트에 객체가 존재하면 true를, 아니면 false를 리턴한다. 

 

 

---

다른 풀이

import java.util.Arrays;
import java.util.stream.IntStream;

class Solution {
    public int[] solution(int[] arr, int k) {
        return IntStream.concat(Arrays.stream(arr).distinct(), IntStream.range(0, k).map(i -> -1)).limit(k).toArray();
    }
}

 

Stream 클래스의 메서드

• concat() : 두 개의 Stream을 전달받아 하나로 합쳐진 새로운 Stream을 반환한다. 
• distinct() : 스트림에서 중복되는 요소들을 모두 제거해 주고 새로운 스트림을 반환한다. 
• limit() : 스트림에서 일정 개수만큼만 데이터를 가져와 새로운 스트림 생성하여 반환한다. 

 

 

 

https://school.programmers.co.kr/learn/courses/30/lessons/181858

프로그래머스