본문 바로가기
알고리즘/[프로그래머스] JAVA

[코딩 기초 트레이닝] 무작위로 K개의 수 뽑기

by 코딩맛집 2024. 2. 5.

 

문제 :

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

해결 :

import java.util.*;

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