aboutsummaryrefslogtreecommitdiff
path: root/challenge-191/ealvar3z/python/ch-2.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-191/ealvar3z/python/ch-2.py')
-rw-r--r--challenge-191/ealvar3z/python/ch-2.py39
1 files changed, 39 insertions, 0 deletions
diff --git a/challenge-191/ealvar3z/python/ch-2.py b/challenge-191/ealvar3z/python/ch-2.py
new file mode 100644
index 0000000000..e013d84da8
--- /dev/null
+++ b/challenge-191/ealvar3z/python/ch-2.py
@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+import unittest
+
+
+class TestSolutionII(unittest.TestCase):
+
+ def cute_list(self, n):
+ bitset_total = 2**n
+ bitset = [[0 for i in range(bitset_total)] for j in range(n+1)]
+ # set the base case
+ bitset[0][0] = 1
+
+ # iterate over all of the positions
+ for i in range(1, n+1):
+ # iterate over all of the subsets
+ for j in range(bitset_total):
+ # iterate over all of the numbers w/in each subset
+ for k in range(n):
+ # for each number in the subset
+ # check if the number exists?
+ visit = (j & (1 << k)) # is the bit set?
+ condition_one_holds = ((k+1) % i == 0)
+ condition_two_holds = (i % (k+1) == 0)
+ if visit and condition_one_holds or condition_two_holds:
+ # if one of these conditions hold, and we have seen the
+ # value, then the list is cute.
+ # Grab the count of what we've seen and return it.
+ # if this bit was set, unset it
+ visited = (j ^ (1 << k))
+ saw_it = bitset[i-1][visited]
+ count = bitset[i][j] = bitset[i][j] + saw_it
+ return count
+
+ def test_cute_list(self):
+ self.assertEqual(self.cute_list(2), 2)
+
+
+if __name__ == "__main__":
+ unittest.main(verbosity=True)