diff options
| -rw-r--r-- | challenge-252/steven-wilson/python/ch-01.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/challenge-252/steven-wilson/python/ch-01.py b/challenge-252/steven-wilson/python/ch-01.py new file mode 100644 index 0000000000..1beb425462 --- /dev/null +++ b/challenge-252/steven-wilson/python/ch-01.py @@ -0,0 +1,23 @@ +#!/usr/bin/env python3 + + +def special_numbers(*integers): + '''Given an array of integers, find the sum of the squares of all special + elements of the array. + + An element $int[i] of @ints is called special if i divides n, i.e. + n % i == 0. Where n is the length of the given array. The array is + 1-indexed. + >>> special_numbers(1, 2, 3, 4) + 21 + >>> special_numbers(2, 7, 1, 19, 18, 3) + 63 + ''' + len_i = len(integers) + return sum(integers[x-1] * integers[x-1] for x in range(1, len_i+1) if len_i % x == 0) + + +if __name__ == "__main__": + import doctest + + doctest.testmod() |