perf: optimize D101.py

had to do this since i saw the horrifying code
kcomain-wasteland · Nov 9, 2023 · d83815f · d83815f
1 parent 067a2e2
commit d83815f
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/Python/Finished/D101.py b/Python/Finished/D101.py
@@ -6,15 +6,11 @@
 (numbers that start with 5, 6 or 9).
 '''
 a = input()
-aw = a[:1]
+aw = int(a[0])
 
-if aw == '2':
-    print('Fixed')
-elif aw == '3':
-    print('Fixed')
-if aw == '5':
-    print('Mobile')
-elif aw == '6':
-    print('Mobile')
-elif aw == '9':
-    print('Mobile')
+if aw in [2, 3]:
+    print("Fixed")
+elif aw in [5, 6, 9]:
+    print("Mobile")
+else:
+    raise Exception("unexpected value")