import itertools def get_two_max(arr): output = [arr[0], arr[1]] if output[0] < output[1]: output = list(reversed(output)) for i in arr: if i > output[0]: output[1] = output[0] output[0] = i elif i > output[1]: output[1] = i return output for x in itertools.permutations((1, 2, 3)): print(x, get_two_max(x))
Standard input is empty
(1, 2, 3) [3, 2] (1, 3, 2) [3, 3] (2, 1, 3) [3, 2] (2, 3, 1) [3, 3] (3, 1, 2) [3, 3] (3, 2, 1) [3, 3]