diff options
Diffstat (limited to 'challenge-172/sgreen/python/ch-1.py')
| -rwxr-xr-x | challenge-172/sgreen/python/ch-1.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/challenge-172/sgreen/python/ch-1.py b/challenge-172/sgreen/python/ch-1.py new file mode 100755 index 0000000000..125ec64815 --- /dev/null +++ b/challenge-172/sgreen/python/ch-1.py @@ -0,0 +1,36 @@ +#!/usr/bin/env python + +import sys +import math +from itertools import combinations + + +def is_prime(number): + '''Return true or false if the number is a prime''' + if number < 2: + return False + + for i in range(2, int(math.sqrt(number)) + 1): + if number % i == 0: + return False + + # It's a prime + return True + + +def main(m, n): + # Retrieve a list of all prime numbers <= m + primes = [x for x in range(m, 1, -1) if is_prime(x)] + + # Go through each combination of n length, and see if we have a solution + for l in combinations(primes, n): + if sum(l) == m: + print(l, sep=', ') + return + + # It is possible that no solution is found + print('No solution!') + + +if __name__ == '__main__': + main(int(sys.argv[1]), int(sys.argv[2])) |
