mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
Added solution for euler problem 493 (#5573)
* Added solution for problem 493 * fixed typo * return result as string
This commit is contained in:
parent
f4fd147d03
commit
0f015fa034
0
project_euler/problem_493/__init__.py
Normal file
0
project_euler/problem_493/__init__.py
Normal file
53
project_euler/problem_493/sol1.py
Normal file
53
project_euler/problem_493/sol1.py
Normal file
|
@ -0,0 +1,53 @@
|
||||||
|
"""
|
||||||
|
Project Euler Problem 493: https://projecteuler.net/problem=493
|
||||||
|
|
||||||
|
70 coloured balls are placed in an urn, 10 for each of the seven rainbow colours.
|
||||||
|
What is the expected number of distinct colours in 20 randomly picked balls?
|
||||||
|
Give your answer with nine digits after the decimal point (a.bcdefghij).
|
||||||
|
|
||||||
|
-----
|
||||||
|
|
||||||
|
This combinatorial problem can be solved by decomposing the problem into the
|
||||||
|
following steps:
|
||||||
|
1. Calculate the total number of possible picking cominations
|
||||||
|
[combinations := binom_coeff(70, 20)]
|
||||||
|
2. Calculate the number of combinations with one colour missing
|
||||||
|
[missing := binom_coeff(60, 20)]
|
||||||
|
3. Calculate the probability of one colour missing
|
||||||
|
[missing_prob := missing / combinations]
|
||||||
|
4. Calculate the probability of no colour missing
|
||||||
|
[no_missing_prob := 1 - missing_prob]
|
||||||
|
5. Calculate the expected number of distinct colours
|
||||||
|
[expected = 7 * no_missing_prob]
|
||||||
|
|
||||||
|
References:
|
||||||
|
- https://en.wikipedia.org/wiki/Binomial_coefficient
|
||||||
|
"""
|
||||||
|
|
||||||
|
import math
|
||||||
|
|
||||||
|
BALLS_PER_COLOUR = 10
|
||||||
|
NUM_COLOURS = 7
|
||||||
|
NUM_BALLS = BALLS_PER_COLOUR * NUM_COLOURS
|
||||||
|
|
||||||
|
|
||||||
|
def solution(num_picks: int = 20) -> str:
|
||||||
|
"""
|
||||||
|
Calculates the expected number of distinct colours
|
||||||
|
|
||||||
|
>>> solution(10)
|
||||||
|
'5.669644129'
|
||||||
|
|
||||||
|
>>> solution(30)
|
||||||
|
'6.985042712'
|
||||||
|
"""
|
||||||
|
total = math.comb(NUM_BALLS, num_picks)
|
||||||
|
missing_colour = math.comb(NUM_BALLS - BALLS_PER_COLOUR, num_picks)
|
||||||
|
|
||||||
|
result = NUM_COLOURS * (1 - missing_colour / total)
|
||||||
|
|
||||||
|
return f"{result:.9f}"
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
print(solution(20))
|
Loading…
Reference in New Issue
Block a user