본문 바로가기
프로그래머스/프로그래머스 Lv0

Lv0 무작위로 K개의 수 뽑기

by 독기품기 2024. 7. 9.

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

 

프로그래머스

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

programmers.co.kr

이 문제는 배열의 원소가 중복되지 않게 k의 길이만큼 배열을 만들어 return하는 것이 목적이기에 먼저 새로운 배열 answer를 선언해주고 for i in arr:로 arr 원소들 i로 받고 if len(answer) >= k 라면 조건에 부합한 것이기에 break를 if i not in answer, 즉 배열 arr의 원소가 answer에 존재하지 않다면(중복되지 않았다면) answer.append(i)를 해주면 된다.

또한 문제에서 k의 길이보다 중복된 원소가 없는 배열의 길이가 짧다면 -1를 append 해 k의 길이와 똑같이 만들라고 했기에 while len(answer) < k를 통해 answer.append(-1)를 반복해주고 return answer를 해주면 된다.