본문 바로가기

코딩테스트

무작위로 K개의 수 뽑기 JAVA 코딩테스트 프로그래머스

무작위로 K개의 수 뽑기 JAVA 

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

 

프로그래머스

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

programmers.co.kr

 

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

 

테스트케이스에서는 통과가 되는데, 제출만 하면 빵점!!!! 결국 구글 검색을 통하여 해결했다.

 

도움받은 블로그 : https://velog.io/@jayking_/%ED%94%84%EB%A1%9C%EA%B7%B8%EB%9E%98%EB%A8%B8%EC%8A%A4-%EB%AC%B4%EC%9E%91%EC%9C%84%EB%A1%9C-K%EA%B0%9C%EC%9D%98-%EC%88%98-%EB%BD%91%EA%B8%B0

 

프로그래머스 - 무작위로 K개의 수 뽑기

풀이풀이방법1) 중복된 값을 제거하기 위한 HashSet 객체 set을 생성2) 중복값이 없을경우 if수행, count가 k와 같아지면 반복문 탈출3) 2번째 반복문을 통해 answer의 빈자리를 -1로 채운다.배열에서 중

velog.io

 

import java.util.*;
    
class Solution {
    public int[] solution(int[] arr, int k) {
        Set<Integer> set = new HashSet<>();
        int[] result = new int[k];
        int idx = 0;

        for(int i=0; i<arr.length; i++){
            if(!set.contains(arr[i])){
                set.add(arr[i]);
                result[idx] = arr[i];
                idx++;
            }
            if(idx == k){
                break;
            }
        }
        
        
       for(int i=idx; i<k; i++){
            result[i] = -1;
        }

        
       //System.out.println(Arrays.toString(result));
         return result;
    }
}

 

728x90
300x250