PE 205 9vs6 dice.py

#number of ways of rolling a total of n using 9 four-sided dice (pyramidal)
p = [0, 0,  0,  1,   9,  45, 165, 486, 1206, 2598, 4950, 8451, 13051, 18351, 23607,\
27876, 30276, 30276, 27876, 23607, 18351, 13051, 8451, 4950, 2598, 1206, 486, 165, 45, 9, 1]
#number of ways of rolling a total of n using 6 six-sided dice (cubic)
c = [1, 6, 21, 56, 126, 252, 456, 756, 1161, 1666, 2247, 2856, 3431, 3906, 4221,\
4332, 4221, 3906, 3431, 2856, 2247, 1666, 1161, 756, 456, 252, 126, 56, 21, 6, 1]
 
w = 0
t = (4**9) * (6**6.)
for C in range(0, 31):
    for P in range (C+1, 31):
        w += p[P]*c[C]
 
print ("Answer to PE205 = {:.7f}".format(w/t))