aboutsummaryrefslogtreecommitdiff
path: root/challenge-214/roger-bell-west/python
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-214/roger-bell-west/python')
-rwxr-xr-xchallenge-214/roger-bell-west/python/ch-1.py44
-rwxr-xr-xchallenge-214/roger-bell-west/python/ch-2.py53
2 files changed, 97 insertions, 0 deletions
diff --git a/challenge-214/roger-bell-west/python/ch-1.py b/challenge-214/roger-bell-west/python/ch-1.py
new file mode 100755
index 0000000000..7eaa04b4f5
--- /dev/null
+++ b/challenge-214/roger-bell-west/python/ch-1.py
@@ -0,0 +1,44 @@
+#! /usr/bin/python3
+
+import unittest
+
+def rankscore(a):
+ av = dict()
+ for s in a:
+ if s in av:
+ av[s] += 1
+ else:
+ av[s] = 1
+ kk = list(av.keys())
+ kk.sort()
+ kk.reverse()
+ rank = 1
+ tab = dict()
+ for k in kk:
+ siz = av[k]
+ r = ""
+ if rank <= 3:
+ r = ["G", "S", "B"][rank - 1]
+ else:
+ r = str(rank)
+ if siz > 1:
+ r += "="
+ tab[k] = r
+ rank += siz
+ return [tab[i] for i in a]
+
+class TestRankscore(unittest.TestCase):
+
+ def test_ex1(self):
+ self.assertEqual(rankscore([1, 2, 4, 3, 5]), ["5", "4", "S", "B", "G"], 'example 1')
+
+ def test_ex2(self):
+ self.assertEqual(rankscore([8, 5, 6, 7, 4]), ["G", "4", "B", "S", "5"], 'example 2')
+
+ def test_ex3(self):
+ self.assertEqual(rankscore([3, 5, 4, 2]), ["B", "G", "S", "4"], 'example 3')
+
+ def test_ex4(self):
+ self.assertEqual(rankscore([2, 5, 2, 1, 7, 5, 1]), ["4=", "S=", "4=", "6=", "G", "S=", "6="], 'example 4')
+
+unittest.main()
diff --git a/challenge-214/roger-bell-west/python/ch-2.py b/challenge-214/roger-bell-west/python/ch-2.py
new file mode 100755
index 0000000000..bb5f35509a
--- /dev/null
+++ b/challenge-214/roger-bell-west/python/ch-2.py
@@ -0,0 +1,53 @@
+#! /usr/bin/python3
+
+from copy import deepcopy
+
+def collectpoints(a):
+ m = []
+ st = 0
+ while st < len(a):
+ k = a[st]
+ e = st
+ while e < len(a) and a[e] == k:
+ e += 1
+ m.append([k, e - st])
+ st = e
+ rv = 0
+ stack = [[m, 0]]
+ while len(stack) > 0:
+ s = stack.pop()
+ if len(s[0]) == 0:
+ rv = max(rv, s[1])
+ else:
+ for i in range(len(s[0])):
+ ss = deepcopy(s)
+ ss[1] += ss[0][i][1] * ss[0][i][1]
+ del ss[0][i]
+ j = i
+ while True:
+ if j > 0 and j < len(ss[0]) and ss[0][j][0] == ss[0][j - 1][0]:
+ ss[0][j][1] += ss[0][j - 1][1]
+ del ss[0][j - 1]
+ j -= 1
+ else:
+ break
+ stack.append(ss)
+ return rv
+
+import unittest
+
+class TestCollectpoints(unittest.TestCase):
+
+ def test_ex1(self):
+ self.assertEqual(collectpoints([2, 4, 3, 3, 3, 4, 5, 4, 2]), 23, 'example 1')
+
+ def test_ex2(self):
+ self.assertEqual(collectpoints([1, 2, 2, 2, 2, 1]), 20, 'example 2')
+
+ def test_ex3(self):
+ self.assertEqual(collectpoints([1]), 1, 'example 3')
+
+ def test_ex4(self):
+ self.assertEqual(collectpoints([2, 2, 2, 1, 1, 2, 2, 2]), 40, 'example 4')
+
+unittest.main()