-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpermutation.py
More file actions
51 lines (43 loc) · 1.18 KB
/
permutation.py
File metadata and controls
51 lines (43 loc) · 1.18 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import unittest
def swap(arr, a, b):
arr[a], arr[b] = arr[b], arr[a]
def permutation(arr, length, offset, result):
if offset == length:
result.append(arr[:])
return
for i in range(offset, length):
swap(arr, i, offset)
permutation(arr, length, offset + 1, result)
swap(arr, i, offset)
class TestPermutation(unittest.TestCase):
def test(self):
arr = ['A', 'B', 'C']
result = []
permutation(arr, len(arr), 0, result)
self.assertEqual(
result,
[
['A', 'B', 'C'],
['A', 'C', 'B'],
['B', 'A', 'C'],
['B', 'C', 'A'],
['C', 'B', 'A'],
['C', 'A', 'B']
]
)
arr = ['1', '2', '3']
result = []
permutation(arr, len(arr), 0, result)
self.assertEqual(
result,
[
['1', '2', '3'],
['1', '3', '2'],
['2', '1', '3'],
['2', '3', '1'],
['3', '2', '1'],
['3', '1', '2']
]
)
if __name__ == '__main__':
unittest.TestCase()