mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-30 16:31:08 +00:00
27 lines
619 B
Python
27 lines
619 B
Python
|
def permute(nums: list[int]) -> list[list[int]]:
|
||
|
"""
|
||
|
Return all permutations.
|
||
|
|
||
|
>>> from itertools import permutations
|
||
|
>>> numbers= [1,2,3]
|
||
|
>>> all(list(nums) in permute(numbers) for nums in permutations(numbers))
|
||
|
True
|
||
|
"""
|
||
|
result = []
|
||
|
if len(nums) == 1:
|
||
|
return [nums.copy()]
|
||
|
for _ in range(len(nums)):
|
||
|
n = nums.pop(0)
|
||
|
permutations = permute(nums)
|
||
|
for perm in permutations:
|
||
|
perm.append(n)
|
||
|
result.extend(permutations)
|
||
|
nums.append(n)
|
||
|
return result
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
import doctest
|
||
|
|
||
|
doctest.testmod()
|