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

[프로그래머스] Lv1 - 포켓몬 (파이썬)

by 2HS 2022. 3. 18.

https://programmers.co.kr/learn/courses/30/lessons/1845

 

코딩테스트 연습 - 폰켓몬

당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.

programmers.co.kr

 

def solution(nums):
    pick = len(nums) // 2
    nums = set(nums)
    answer = pick if len(nums) >= pick else len(nums)
    return answer

 

풀이
1. 골라야 할 개수 pick 정의
2. 집합의 성질을 이용해서 중복되는 수 제거
3. 중복된 수를 제거한 리스트의 길이가 골라야 할 개수보다 많거나 같으면 (len(nums) >= pick) pick
4. 아니면 len(nums)