프로그래머스/기초
[프로그래머스] Lv.0 /무작위로 K개의 수 뽑기 (파이썬/Python)
junslee
2025. 3. 7. 13:10
1. 문제 설명
2. 풀이 과정
결과 result를 빈 문자열로 받고, seen을 set()으로 초기화한다.
arr의 문자 num을 for문으로 반복해서
num이 seen에 없다면, result에 num을 append()하고, seen에 num을 add()한다.
그리고 result의 길이가 k랑 같다면 break로 반복문을 빠져나온다.
while문으로 result의 길이len()이 k보다 작다면 appned(-1)으로 -1을 추가하는 것을 반복한다.
최종적으로 result를 반환한다.
3. 코드
def solution(arr, k):
result = []
seen = set()
for num in arr:
if num not in seen:
result.append(num)
seen.add(num)
if len(result) == k:
break
while len(result) < k:
result.append(-1)
return result