aboutsummaryrefslogtreecommitdiff
path: root/challenge-215/sgreen/python/ch-2.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-215/sgreen/python/ch-2.py')
-rwxr-xr-xchallenge-215/sgreen/python/ch-2.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/challenge-215/sgreen/python/ch-2.py b/challenge-215/sgreen/python/ch-2.py
new file mode 100755
index 0000000000..9fc8a9a7e9
--- /dev/null
+++ b/challenge-215/sgreen/python/ch-2.py
@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+
+import sys
+
+
+def main(array):
+ to_place = array.pop()
+
+ for i in range(1, len(array)-1):
+ # Are the number before the current one, the current one and the next one all zero?
+ if array[i-1] == 0 and array[i] == 0 and array[i+1] == 0:
+ # If so, we can put a one here
+ array[i] = 1
+ to_place -= 1
+ if to_place == 0:
+ # We have placed all the ones
+ print(1)
+ return
+
+ # There is no possible solution
+ print(0)
+
+
+if __name__ == '__main__':
+ # Turn the strings into integers
+ n = [int(i) for i in sys.argv[1:]]
+ main(n)