From 04322e67e5d3754f7be6525c0e078a955237067a Mon Sep 17 00:00:00 2001
From: Guillaume Rochedix <33205665+grochedix@users.noreply.github.com>
Date: Tue, 29 Sep 2020 12:38:12 +0200
Subject: [PATCH] Heaps algorithm iterative (#2505)

* heap's algorithm iterative

* doctest

* doctest

* rebuild
---
 .../heaps_algorithm_iterative.py              | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 divide_and_conquer/heaps_algorithm_iterative.py

diff --git a/divide_and_conquer/heaps_algorithm_iterative.py b/divide_and_conquer/heaps_algorithm_iterative.py
new file mode 100644
index 000000000..4dab41f53
--- /dev/null
+++ b/divide_and_conquer/heaps_algorithm_iterative.py
@@ -0,0 +1,60 @@
+"""
+Heap's (iterative) algorithm returns the list of all permutations possible from a list.
+It minimizes movement by generating each permutation from the previous one
+by swapping only two elements.
+More information:
+https://en.wikipedia.org/wiki/Heap%27s_algorithm.
+"""
+
+
+def heaps(arr: list) -> list:
+    """
+    Pure python implementation of the iterative Heap's algorithm,
+    returning all permutations of a list.
+    >>> heaps([])
+    [()]
+    >>> heaps([0])
+    [(0,)]
+    >>> heaps([-1, 1])
+    [(-1, 1), (1, -1)]
+    >>> heaps([1, 2, 3])
+    [(1, 2, 3), (2, 1, 3), (3, 1, 2), (1, 3, 2), (2, 3, 1), (3, 2, 1)]
+    >>> from itertools import permutations
+    >>> sorted(heaps([1,2,3])) == sorted(permutations([1,2,3]))
+    True
+    >>> all(sorted(heaps(x)) == sorted(permutations(x))
+    ...     for x in ([], [0], [-1, 1], [1, 2, 3]))
+    True
+    """
+
+    if len(arr) <= 1:
+        return [tuple(arr)]
+
+    res = []
+
+    def generate(n: int, arr: list):
+        c = [0] * n
+        res.append(tuple(arr))
+
+        i = 0
+        while i < n:
+            if c[i] < i:
+                if i % 2 == 0:
+                    arr[0], arr[i] = arr[i], arr[0]
+                else:
+                    arr[c[i]], arr[i] = arr[i], arr[c[i]]
+                res.append(tuple(arr))
+                c[i] += 1
+                i = 0
+            else:
+                c[i] = 0
+                i += 1
+
+    generate(len(arr), arr)
+    return res
+
+
+if __name__ == "__main__":
+    user_input = input("Enter numbers separated by a comma:\n").strip()
+    arr = [int(item) for item in user_input.split(",")]
+    print(heaps(arr))