반응형
https://www.acmicpc.net/problem/13251
13251번: 조약돌 꺼내기
첫째 줄에 뽑은 조약돌이 모두 같은 색일 확률을 출력한다. 정답과의 절대/상대 오차는 10-9까지 허용한다.
www.acmicpc.net
import sys
import math
M = int(sys.stdin.readline())
stone = list(map(int,sys.stdin.readline().split()))
K = int(sys.stdin.readline())
N = sum(stone)
total = math.comb(N, K)
same_color = 0
for s in stone:
same_color += math.comb(s, K)
print(same_color/total)반응형