080 조약돌 꺼내기

첫번쨰 코드

 

이번에는 단순하기 조합으로 풀기보다는 조금 더 생각해서

점화식을 생각한 후에 해당 방법으로 코드를 작성해야 한다.

 

아이디어는 책과 동일하게 생각했지만,

점화식을 코드로 구현하는 방법에서 책을 참고하였다.

 

import sys
input = sys.stdin.readline

probability = [0] * 51
m = int(input())
colors = list(map(int, input().split()))
total = 0

for i in colors:
    total += i

k = int(input())
answer = 0

for i in range(m):
    if colors[i] >= k:
        probability[i] = 1
        for j in range(k):
            probability[i] = probability[i] * (colors[i] - j) / (total - j)
        answer += probability[i]

print(answer)

 

통과

---

링크

https://github.com/ornni/programmers/tree/main/%EB%B0%B1%EC%A4%80/Silver/13251.%E2%80%85%EC%A1%B0%EC%95%BD%EB%8F%8C%E2%80%85%EA%BA%BC%EB%82%B4%EA%B8%B0

programmers/백준/Silver/13251. 조약돌 꺼내기 at main · ornni/programmers