티스토리 뷰

728x90

문제

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

 

코딩테스트 연습 - 폰켓몬

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

programmers.co.kr

 

풀이 및 소스코드

굉장히 쉬운 문제다.

set 함수를 사용해 중복없이 nums를 새로운 배열에 담은 다음에, 그 배열의 길이가 n/2 보다 길다면 n/2를 answer에 담고, 짧다면 배열의길이를 answer에 담으면 된다.

def solution(nums):
    answer = 0
    s_nums = set(nums)
    if (len(s_nums)>=(len(nums)//2)):
        answer = len(nums)//2
    else:
        answer = len(s_nums)
    return answer
반응형