[프로그래머스] 무작위로 K개의 수 뽑기 (stream.distinct()) - JAVA

프로그래머스 _ 코딩 기초 트레이닝 DAY 19 - (5)

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

프로그래머스

 

문제 | 랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.정수 배열 arr가 주어집니다. 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

 

입력 #1 | arr = [0, 1, 1, 1, 1], k = 4

출력 #1 | [0, 1, -1, -1]

---

>> 풀이 << 

우선 arr의 중복을 제거한 후, arr의 배열 길이와 k 값을 비교하는 것을 목표로 설정했다. 아예 크기가 k인 배열을 선언한 후, arr의 중복을 제거한 배열 값을 넣고 남는 자리에 -1을 채워넣는 방법으로 문제풀이를 진행했다!

 

자바를 공부했을 때, 책에서 봤던 stream의 distinct 메소드가 생각나서 그걸 사용해보기로 했다.

 

•   스트림의 중복 제거하기 - distinct()
  • Stream stream().distinct()는 주어진 스트림에서의 중복을 제거해 새로운 스트림을 반환해주는 메소드이다.
  • 스트림 말고 예를 들어 List로 변환하고 싶다면 collect() 메소드를 사용하면 된다.

---

코드는 다음과 같다.

import java.util.*;
class Solution {
    public int[] solution(int[] arr, int k) {
        int[] ans = Arrays.stream(arr).distinct().toArray();
        int[] answer = new int[k];
        
        for (int i = 0 ; i < k ; i++) {
            if (i >= ans.length) answer[i] = -1;
            else answer[i] = ans[i];
        }
        return answer;
    }
}