프로그래머스 문제기록

[프로그래머스 코딩테스트 입문] - [가위바위보, 구슬 나누는 경우의 수]

코딩펭귄 2023. 9. 15. 21:41

가위바위보  

 문제설명 

가위=2, 바위=0, 보=5 

가위 바위 보를 내는 순서대로 나타낸 문자열 rsp가 매개변수로 주어짐

rsp에 저장된 가위바위보를 모두 이기는 경우를 순서대로 나타낸 문자열을 return

 입출력 예 

 rsp  |  result
 "2"       "0"    -> "2"는 가위이므로 바위를 나타내는 "0"을 return 
"205"  "052"  -> "205"는 순서대로 가위, 바위, 보이고 이를 모두 이기려면 바위, 보, 가위를 순서대로 내야하므로 “052”를 return

 코드 

(풀이1)
def solution(rsp):
    answer = ''
    for i in rsp:
        if i == '2':
            answer += '0'
        elif i == '0':
            answer += '5'
        else:
            answer += '2'  
    return answer
   
   
 (풀이2)   
 def solution(rsp):
 	 d = {'0':'5', '2':'0', '5':'2'}   
     return ''.join(d[i] for i in rsp)

 

구슬을 나누는 경우의 수

 문제설명 

매개변수 : 구슬의개수 : balls  (1 ≤ balls ≤ 30) / 친구들에게 나눠줄 구슬의 개수 : share (1 ≤ share ≤ 30)

balls 개의 구슬 중  share개의 구슬을 고르는 가능한 모든 경우의수 return

 예시 

balls | share | result
   3         2        3
   5         3       10

 힌트 

서로 다른 n개 중 m개를 뽑는 경우의 수 공식 -> n! / ((n-m)!*m!)

1) math함수 쓸때

import math 

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

comb(n,k) = nCk

- n개의 수 중 k개를 꺼내는 수 (n,k는 모두 int값이어야 함)

 

2) math함수 안쓸때

def facto(num):
    i = 1
    for j in range(1, num+1):
        i *= j
    return i

def solution(balls, share):
    answer=0
    num1 = facto(balls)
    num2 = facto(balls-share)
    num3 = facto(share)
    return num1/(num2*num3)