🤖 옌지니어의 IT 기록 🤖

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

문제

문제 설명

머쓱이는 구슬을 친구들에게 나누어주려고 합니다. 구슬은 모두 다르게 생겼습니다. 머쓱이가 갖고 있는 구슬의 개수 balls와 친구들에게 나누어 줄 구슬 개수 share이 매개변수로 주어질 때, balls개의 구슬 중 share개의 구슬을 고르는 가능한 모든 경우의 수를 return 하는 solution 함수를 완성해주세요.

제한 사항

• 1 ≤ balls ≤ 30
• 1 ≤ share ≤ 30
• 구슬을 고르는 순서는 고려하지 않습니다.
• share ≤ balls

입출력 예

소스 코드

# 1 (런타임 에러)
def solution(balls, share):
    answer = 1
    
    ball_list = list(range(balls,1,-1))
    share_list = list(range(1,share+1))
    
    for idx in range(1,share+1):
        answer *= ball_list[idx-1]
        answer = int(answer/share_list[idx-1])
    
    
    return answer

풀이 # 1

• 공의 갯수에서 1씩 빼서 ball_list에 담고, 공유할 공의 갯수를 1부터 1씩 더해 share_list에 담기
• share 숫자 범위에서 ball_list의 원소는 곱하고 share_list의 원소는 나누기

# 2
def solution(balls, share):
    answer = 1
    cnt = 0

    while cnt < share:
        answer *= (balls-cnt)
        answer = int(answer/(cnt+1))
        cnt += 1    
    
    return answer

풀이 #2

• cnt가 share의 숫자보다 작을동안 곱하기와 나누기를 실행
  • balls-cnt를 곱하기
  • 위의 숫자에서 cnt+1을 나누기 

# 3
import math

def solution(balls, share):
    return math.comb(balls, share)

풀이 #3

• math 라이브러리에서 combination 함수 사용

처음에 숫자를 곱하고 나누는 방향으로 하다가 잘 안되길래 리스트로 풀어봤는데 역시나 런타임 에러가 났다. 그래서 숫자를 조작해서 while문에 넣었더니 결과는 나왔지만 뭔가 더 빠르게 나왔으면 좋겠다는 생각이 들었다. 다른 사람들의 풀이를 보다보니 math 라이브러리에 combination 함수가 있었다...라이브러리 함수 쓰는거 최고...!!