aboutsummaryrefslogtreecommitdiff
path: root/challenge-258/packy-anderson/python/ch-2.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-258/packy-anderson/python/ch-2.py')
-rwxr-xr-xchallenge-258/packy-anderson/python/ch-2.py34
1 files changed, 34 insertions, 0 deletions
diff --git a/challenge-258/packy-anderson/python/ch-2.py b/challenge-258/packy-anderson/python/ch-2.py
new file mode 100755
index 0000000000..b9bdce3bb0
--- /dev/null
+++ b/challenge-258/packy-anderson/python/ch-2.py
@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+
+from functools import cache
+
+@cache
+def setBitCount(i):
+ count = 0
+ bit = 1
+ while (bit <= i):
+ if i & bit: count += 1 # count if we have this bit set
+ bit <<= 1 # shift bits left, ie 10 becomes 100
+ return count
+
+def valueSum(k, ints):
+ sum = 0
+ for i in range(len(ints)):
+ if setBitCount(i) == k: sum += ints[i]
+ return sum
+
+def comma_join(arr):
+ return ', '.join(map(lambda i: str(i), arr))
+
+def solution(k, ints):
+ print(f'Input: @ints = ({comma_join(ints)}), $k = {k}')
+ print(f'Output: {valueSum(k, ints)}')
+
+print('Example 1:')
+solution(1, [2, 5, 9, 11, 3])
+
+print('\nExample 2:')
+solution(2, [2, 5, 9, 11, 3])
+
+print('\nExample 3:')
+solution(0, [2, 5, 9, 11, 3])