문제 설명
랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다. 적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.
이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.
정수 배열 arr가 주어집니다. 문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.
단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.
제한사항
1 ≤ arr의 길이 ≤ 100,000
- 0 ≤ arr의 원소 ≤ 100,000
1 ≤ k ≤ 1,000
입출력 예
arr | k | result |
[0,1,1,2,2,3] | 3 | [0,1,2] |
[0,1,1,1,1] | 4 | [0,1,-1,-1] |
입출력 예 설명
입출력 예 #1
앞에서부터 서로 다른 k개의 수를 골라내면 [0, 1, 2]가 됩니다. 따라서 [0, 1, 2]를 return 합니다.
입출력 예 #2
서로 다른 수의 개수가 2개 뿐이므로 서로 다른 수들을 앞에서부터 차례대로 저장한 [0, 1]에서 이후 2개의 인덱스를 -1로 채워넣은 [0, 1, -1, -1]을 return 합니다.
풀이
-> 동적으로 배열을 담기 위해 우선 ArrayList를 사용하였다. answer의 크기는 k로 초기화하였고, -1값을 기본적으로 담아두었다. 그 다음 ArrayList인 temp를 선언하여 temp에 arr[i] 값이 없으면 추가하였다. temp에 값을 다 추가한 후에 temp의 size가 큰지, k 값이 큰지 삼항연산자를 통해 비교하여 answer에 temp값을 담는 for문에 사용하였다. 그런 후에 return~
코드
import java.util.*;
class Solution {
public int[] solution(int[] arr, int k) {
int[] answer = {};
answer = new int[k];
for(int i=0;i<k;i++){
answer[i]=-1;
}
ArrayList<Integer> temp = new ArrayList<>();
for(int i=0;i<arr.length;i++){
if(!temp.contains(arr[i])){
temp.add(arr[i]);
}
}
int n = temp.size()>k?k:temp.size();
for(int i=0;i<n;i++){
answer[i]=temp.get(i);
}
return answer;
}
}
'Algorithm > Programmers 입문' 카테고리의 다른 글
[프로그래머스 코딩기초 트레이닝] 전국 대회 선발 고사(JAVA) (1) | 2024.01.11 |
---|---|
[프로그래머스 코딩기초 트레이닝] 문자열 묶기(JAVA) (1) | 2024.01.07 |
[프로그래머스 코딩기초 트레이닝] 문자열 바꿔서 찾기 (JAVA) (0) | 2023.12.27 |
[프로그래머스 코딩기초 트레이닝] A 강조하기 (JAVA) (1) | 2023.12.07 |
[프로그래머스 코딩기초 트레이닝] 조건에 맞게 수열 변환하기 2 (JAVA) (1) | 2023.12.07 |