프로그래머스 문제기록
[프로그래머스 코딩테스트 입문] - [가위바위보, 구슬 나누는 경우의 수]
코딩펭귄
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)