Merge pull request #4731 from stuart-little/stuart-little_126_python

1st commit on 126_python

2 files changed, 33 insertions, 0 deletions

diff --git a/challenge-126/stuart-little/python/ch-1.py b/challenge-126/stuart-little/python/ch-1.py
new file mode 100755
index 0000000000..aea89f2b33
--- /dev/null
+++ b/challenge-126/stuart-little/python/ch-1.py
@@ -0,0 +1,14 @@
+#!/usr/bin/env python
+
+# run <script> <number>
+
+import sys
+
+def no1(nr):
+    if len(nr)==0:
+        return 0
+    if nr[0] == '1':
+        return 9**(len(nr)-1)
+    return (int(nr[0])-1) * 9**(len(nr)-1) + no1(nr[1:])
+
+print(no1(sys.argv[1]) if '1' not in sys.argv[1] else (no1(sys.argv[1])-1))
diff --git a/challenge-126/stuart-little/python/ch-2.py b/challenge-126/stuart-little/python/ch-2.py
new file mode 100755
index 0000000000..971e35a639
--- /dev/null
+++ b/challenge-126/stuart-little/python/ch-2.py
@@ -0,0 +1,19 @@
+#!/usr/bin/env python
+
+# run <script>
+
+def nbrs(mat, i, j):
+    return [(i+x,j+y) for x in range(-1,2) for y in range(-1,2) if (x != 0 or y != 0) and (0 <= i+x < len(mat)) and (0 <= j+y < len(mat[0]))]
+
+inptStr="""x * * * x * x x x x
+* * * * * * * * * x
+* * * * x * x * x *
+* * * x x * * * * *
+x * * * x * * * * x
+"""
+inpt = [s.split() for s in inptStr.splitlines()]
+
+for i in range(len(inpt)):
+    for j in range(len(inpt[0])):
+        print(inpt[i][j] if (inpt[i][j] == 'x') else len(list(filter(lambda p: inpt[p[0]][p[1]] == 'x', nbrs(inpt,i,j)))), end=" ")
+    print()