Removed infinite loop from Nth magic no.

Algo-Lab · Oct 9, 2018 · fbee54a · fbee54a
1 parent 7087bdf
commit fbee54a
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/BitManipulation/Nth_magic_number/nth_magic_number.cpp b/BitManipulation/Nth_magic_number/nth_magic_number.cpp
@@ -1,25 +1,29 @@
 #include <bits/stdc++.h> 
 using namespace std; 
-  
+
 int nthMagicNo(int n) 
 { 
     int pow = 1, answer = 0; 
-
+
+
     while (n) 
     { 
-       pow = pow*5; 
-
-       if (n & 1) 
-         answer += pow; 
-
+    pow = pow*5; 
+
+
+    if (n & 1) 
+        answer += pow; 
+
+
+    n >>= 1;
     } 
     return answer; 
 } 
-  
+
 int main() 
 { 
     int n;
-    cin>>n; 
+    cin>>n;
     cout << "nth magic number is " << nthMagicNo(n) << endl; 
     return 0; 
-} 
+}