프로그래머스-폰켓몬

폰켓몬

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

코딩테스트 연습 - 폰켓몬

 

처음에 이 문제 이해를 제대로 못해서 폰켓몬 종류를 고르는 경우의 수 조합을 구하는 줄 알았다. 그래서 itertools 라이브러리를 써서 combination 함수를 통해 조합을 구했는데, 오답이었다.

그래서 문제를 다시 잘 읽어보니 result 값으로 최대로 고를 수 있는 폰켓몬 종류의 수를 구하는 문제였다.

특히 이 부분을 유심히 읽어봐야 했다.

 

• 가장 많은 종류의 폰켓몬을 선택하는 방법이 여러 가지인 경우에도, 선택할 수 있는 폰켓몬 종류 개수의 최댓값 하나만 return 하면 됩니다.
• 최대한 많은 종류의 폰켓몬을 포함해서 N/2마리를 선택하려 합니다.

def solution(nums):
    length=int(len(nums)/2)
    nums=set(nums)
    answer=len(nums)
    if answer>length:
        answer=length
    return answer

 

위의 조건을 바탕으로 코드를 짜면 위와 같다.

length 변수에 N/2 만큼의 수를 넣어준 뒤에, set 집합 자료형을 이용해서 nums에 들어 있는 중복 원소를 제거한다. 그런 다음 nums의 길이를 구한 다음, 만약에 nums의 길이가 N/2 수보다 크면, 최대 N/2마리까지 고를 수 있기 때문에 N/2로 바꿔준다.

문제를 풀고나서 다른 사람 풀이 보러 들어갔는데….

 

def solution(ls):
    return min(len(ls)/2, len(set(ls)))

기가 막힌다. min함수를 쓰면 N/2랑 중복을 제거한 nums길이중에서 더 작은 것을 반환하면 되니까 이렇게 작성한거다. 감탄만 나온다.