Update permute and add dft.py

README.md

@@ -74,8 +74,8 @@ Some pictures and ideas are from `<<Introduction to Algotithm>>`
         * [isBipartGraph.py](./graph/isBipartGraph.py)
     * [math](./math)
         * [README.md](./math/README.md)
+        * [dft.py](./math/dft.py)
         * [fastPow.py](./math/fastPow.py)
-        * [fft.py](./math/fft.py)
         * [fibonacci](./math/fibonacci)
         * [numWeight](./math/numWeight)
         * [numberTheory](./math/numberTheory)

math/permute/permute_back_track.py

@@ -9,15 +9,19 @@
 # Description:
 #########################################################################
 '''
-def permute(n):
-    def _util(lst,i):
-        if i==n:print(lst)
-        else:
-            for j in range(i,n):
-                lst[i],lst[j]=lst[j],lst[i]
-                _util(lst,i+1)
-                lst[i],lst[j]=lst[j],lst[i]
-    _util([i for i in range(n)],0)
 
-if __name__=='__main__':
+
+def permute(n):
+    def _util(lst, i):
+        if i == n:
+            print(lst)
+        else:
+            for j in range(i, n):
+                lst[i], lst[j] = lst[j], lst[i]
+                _util(lst, i+1)
+                lst[i], lst[j] = lst[j], lst[i]
+    _util([i for i in range(n)], 0)
+
+
+if __name__ == '__main__':
     permute(5)

math/permute/permute_divide_and_conquer.py

@@ -9,15 +9,19 @@
 # Description:
 #########################################################################
 '''
-def permute(lst,n):
+
+
+def permute(lst, n):
     ''' O(n!), optimal'''
-    if n==1:print(lst)
+    if n == 1:
+        print(lst)
     else:
         for i in range(n):
-            lst[i],lst[n-1] = lst[n-1],lst[i]
-            permute(lst,n-1)
-            lst[i],lst[n-1] = lst[n-1],lst[i]
+            lst[i], lst[n-1] = lst[n-1], lst[i]
+            permute(lst, n-1)
+            lst[i], lst[n-1] = lst[n-1], lst[i]
 
-if __name__=='__main__':
+
+if __name__ == '__main__':
     n = 3
-    permute([i for i in range(n)],n)
+    permute([i for i in range(n)], n)