Merge pull request devAmoghS#3 from luiz-surian/patch-2

Amogh Singhal · web-flow · commit cb713c13f6d7 · 2019-06-07T08:16:31.000+05:30
Update josephus_improved.py
diff --git a/josephus_improved.py b/josephus_improved.py
@@ -12,7 +12,11 @@ def josephus_v2(people, step=2):
     # the loop runs for O(floor(logN)) time
     while p*2 < N:
       p = p*2
-    print((2*(N-p))+1)
+    # If N is a power of 2, should return 1. So let's check if L (N-p) is < p
+    if N-p >= p:
+      print(1)
+    else:
+      print((2*(N-p))+1)
     
 num = int(input("Enter the number of soldiers: "))
 soldiers = [i for i in range(1, num+1)] # generates a list of 1..num
