aboutsummaryrefslogtreecommitdiff
path: root/challenge-052/user-person/python/ch-1.py
diff options
context:
space:
mode:
Diffstat (limited to 'challenge-052/user-person/python/ch-1.py')
-rwxr-xr-xchallenge-052/user-person/python/ch-1.py21
1 files changed, 16 insertions, 5 deletions
diff --git a/challenge-052/user-person/python/ch-1.py b/challenge-052/user-person/python/ch-1.py
index b7a0685098..2cb2de3814 100755
--- a/challenge-052/user-person/python/ch-1.py
+++ b/challenge-052/user-person/python/ch-1.py
@@ -1,7 +1,7 @@
#!/usr/bin/env python3
###########################################################################
-# script name: ch-1.py #
+# script name: ch-1-CORRECTED.py #
# #
# https://github.com/user-person #
# #
@@ -25,17 +25,28 @@ step = []
UPPER_LIMIT = 1000
LOWER_LIMIT = 99
-for i in range(1,8):
- step.append( (i * 100) + ((i+1) * 10) + (i+2) )
+for i in range(1,10):
+ if i < 8: # UP UP
+ step.append( (i * 100) + ((i+1) * 10) + (i+2) )
+
+ if i > 1: # DOWN DOWN
+ step.append( (i * 100) + ((i-1) * 10) + (i-2) )
+
+ if i < 9: # UP DOWN
+ step.append( (i * 100) + (i+1) * 10 + i )
+
+ step.append( (i * 100) + (i-1) * 10 + i ) # DOWN UP
+
+step.sort()
errorString = os.path.basename(sys.argv[0]) + ' requires 2 arguments between 100 and 999.'
if len(sys.argv) != 3:
print(errorString)
exit(1)
-
-args = [int(arg) for arg in sys.argv[1:]]
+args = [int(arg) for arg in sys.argv[1:]]
+
for arg in args:
if arg < LOWER_LIMIT or arg > UPPER_LIMIT:
print(errorString)