[JAVA] 프로그래머스 - 무작위로 k개의 수 뽑기

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

프로그래머스

 

 

 

 

 

정답 

 

import java.util.ArrayList;
import java.util.Arrays;

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

        
        if(list.size()>k)
        {
            for(int i=0; i<k; i++)
            {
                answer[i] = list.get(i);
            }
        }
        else
        {
            for(int i=0; i<list.size(); i++)
            {
                answer[i] = list.get(i);
            }
        }
        
        
        
        return answer;
        
    }
}

 

 

다른 사람의 풀이 

 

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int k) {
        Set<Integer> set = new HashSet<>();
        int[] answer = new int[k];
        for (int i = 0; i < k; i++) {
            answer[i] = -1;
        }
        int count = 0;
        for (int i = 0; i < arr.length; i++) {
            if(!set.contains(arr[i])){
                answer[count++] = arr[i];
                set.add(arr[i]);
            }
            if(count == answer.length){
                break;
            }
        }
        return answer;
    }
}