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

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

프로그래머스

 

• 총 N 마리의 폰켓몬 중에서 N/2마리를 가져갈 수 있음
• 폰켓몬은 종류에 따라 번호를 붙여 구분
• 최대한 많은 종류의 폰켓몬을 포함해서 N/2마리를 선택하려 함
• nums: N마리 폰켓몬의 종류 번호가 담긴 배열
• N/2마리의 폰켓몬을 선택하는 방법 중, 가장 많은 종류의 폰켓몬을 선택하는 방법을 찾아, 그때의 폰켓몬 종류 번호의 개수를 return

제한사항

• nums는 폰켓몬의 종류 번호가 담긴 1차원 배열
• nums의 길이(N)는 1 이상 10,000 이하의 자연수이며, 항상 짝수
• 폰켓몬의 종류 번호는 1 이상 200,000 이하의 자연수
• 가장 많은 종류의 폰켓몬을 선택하는 방법이 여러 가지인 경우에도, 선택할 수 있는 폰켓몬 종류 개수의 최댓값 하나만 return

 

# 총 4마리의 폰켓몬이 있고, 각 폰켓몬의 종류 번호가 [3번, 1번, 2번, 3번]일 때
# -> 3번 폰켓몬 두 마리, 1번 폰켓몬 한 마리, 2번 폰켓몬 한 마리

# 이때, 4마리의 폰켓몬 중 2마리를 고르는 방법
# 1) 첫 번째(3번), 두 번째(1번)
# 2) 첫 번째(3번), 세 번째(2번)
# 3) 첫 번째(3번), 네 번째(3번)
# 4) 두 번째(1번), 세 번째(2번)
# 5) 두 번째(1번), 네 번째(3번)
# 6) 세 번째(2번), 네 번째(3번)
# => 6가지

# -> 이때, 첫 번째(3번) 폰켓몬과 네 번째(3번) 폰켓몬을 선택하는 방법은 한 종류(3번 폰켓몬 두 마리)의 폰켓몬만 가질 수 있지만, 다른 방법들은 모두 두 종류의 폰켓몬을 가질 수 있음
# => 가질 수 있는 폰켓몬 종류 수의 최댓값 = 2

 

"""
입출력 예시)

nums = [3,1,2,3] -> 2
nums = [3,3,3,2,2,4] -> 3
nums = [3,3,3,2,2,2] -> 2
"""

 

 

첫 번째 시도

 

## 의사코드 ##

# N 마리의 폰켓몬 중 N/2 마리를 가져오는 방법의 수 -> combinations(list, N/2)
# (순서 상관 x, 중복 x로 뽑기 때문에 combinations)

 

from itertools import combinations 
def solution(nums):
    answer = 0
    N_2 = len(nums) // 2
    method = list(combinations(nums, N_2))
    for i in range(len(method)):
        if method[i][0] != method[i][1]:
            answer += 1
        else:
            continue
        
    return answer
    
    
 nums = [3,1,2,3]
 solution(nums)
 # -> 5

-> 이건 그냥 다른 종류의 폰켓몬을 고르는 방법의 수를 구한 것이 됨

 

• combinations()
  • 서로 다른 n개에서 r개를 선택할 때, 순서를 고려하지 않고, 중복 없이 뽑는 경우의 수
  • 순서를 고려하지 않는다 => AB = BA
  • ex) combinations('ABCD', 2)
    • => AB, AC, AD, BC, BD, CD

from itertools import combinations
combi = combinations(iterable, r)
print(list(combi))

 

• permutations()
  • 서로 다른 n개에서 r개를 선택할 때, 순서를 고려해서 중복 없이 뽑는 경우의 수 
  • 순서를 고려한다 => AB != BA
  • ex) permutations('ABCD', 2)
    • => AB, AC, AD, BA, BC, BD, CA, CB, CD, DA, DB, DC

from itertools import permutations
permu = permutations(iterable, r)
print(list(permu))

 

❗️combinations()와 permutations() 모두 값을 출력하기 위해서는 list나 tuple 형식으로 바꿔준 후 출력해야 함❗️

 

 

통과한 코드

 

## 의사코드 ##

# 일단 폰켓몬의 종류의 개수를 구함 -> set()
# 가져올 수 있는 개수인 N/2를 구함
# if N/2 <= 폰켓몬의 종류의 개수: (가져올 수 있는 개수(N/2)가 폰켓몬의 종류의 개수보다 작거나 같은 경우)
#    answer = N/2 (가장 많은 종류의 폰켓몬을 선택했을 때의 폰켓몬 종류의 개수는 가져올 수 있는 개수(N/2))
# else: (가져올 수 있는 개수(N/2)가 폰켓몬의 종류의 개수보다 큰 경우)
#   answer = 폰켓몬의 종류의 개수 (가장 많은 종류의 폰켓몬을 선택했을 때의 폰켓몬 종류의 개수는 폰켓몬의 종류의 개수)

 

def solution(nums):
    type_n = len(set(nums)) # 폰켓몬의 종류의 개수
    N_2 = len(nums) // 2 # 가져올 수 있는 개수(N/2)
    if N_2 <= type_n: # 가져올 수 있는 개수(N/2)가 폰켓몬의 종류의 개수보다 작거나 같은 경우
        answer = N_2 # 가장 많은 종류의 폰켓몬을 선택했을 때의 폰켓몬 종류의 개수는 가져올 수 있는 개수(N/2)
    else: # 가져올 수 있는 개수(N/2)가 폰켓몬의 종류의 개수보다 큰 경우
        answer = type_n # 가장 많은 종류의 폰켓몬을 선택했을 때의 폰켓몬 종류의 개수는 폰켓몬의 종류의 개수
    return answer

-> set()을 이용해서 폰켓몬의 종류의 개수를 구한 다음, 폰켓몬의 종류의 개수와 N/2 중 출력

 

 

다른 풀이

 

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

-> len(nums)/2 = N/2, len(set(nums) = 폰켓몬의 종류의 개수

=> N/2와 폰켓몬의 종류의 개수 중 min 값을 출력

 

 

cf) 이전에 작성했던 코드

 

https://monzheld.tistory.com/48

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

 

 

참고)

 

https://pearlluck.tistory.com/468

[Python] 순열(permutations )과 조합(combinations)