[Python] 프로그래머스 Level1 폰켓몬

지애·2024년 6월 24일
1

코딩테스트

목록 보기
1/12

풀이

구해야 하는 대상만 알면 쉬운 문제!
각 폰켓몬마다 고유한 번호를 가진다는 조건을 보고 총 폰켓몬 가짓수를 파악하기 위해 set을 사용했다.

# 선택한 폰켓몬 종류 수의 최댓값 구하기
def solution(nums):
    nums_length = len(nums)
    nums_set = set(nums)
    if nums_length//2 < len(nums_set):
        answer = nums_length//2
    else:
        answer = len(nums_set)
    return answer

다른 풀이 보고 배운 점

    if nums_length//2 < len(nums_set):
        answer = nums_length//2
    else:
        answer = len(nums_set)

이 부분을 if else문이 아닌 min함수를 사용했더라면 더 간결하게 작성할 수 있다.

profile
차근차근

0개의 댓글