[프로그래머스][PYTHON] Lv. 0 무작위로 K개의 수 뽑기

문제 설명

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

프로그래머스

 

 

내 코드

def solution(arr, k):
    # answer = list(set(arr))
    answer = [i for n, i in enumerate(arr) if i not in arr[:n]]
    if len(answer) >= k:
        answer = answer[:k]
    else:
        for x in range(k-len(answer)):
            answer.append(-1)
    return answer

 

처음에는 set을 이용해 문제를 풀었다. 

 

테스트 결과는 모두 통과하여 쉽게 통과할 줄 알았는데, 채점해보니 모든 결과가 틀리게 나왔다!

 

반례를 찾아 등록해보았고, 

 

set 자료형은 원본 리스트의 순서가 유지되지 않아, 원하는 값이 나오지 않는 것을 확인했다. 

따라서 원본리스트의 순서를 유지하며 중복을 제거할 수 있는 방법을 사용하였고, 이는 따로 글로 정리해두었다. 

2024.02.13 - [PYTHON] - [PYTHON] 리스트 중복 제거 & 순서 유지

[PYTHON] 리스트 중복 제거 & 순서 유지

 

끄읏-!