def func(array1, array2): diff = sum(array1) - sum(array2) if diff % 2 == 1: return [] else: diff >>= 1 array2 = set(array2) for i in array1: if i - diff in array2: return [i, i - diff] return [] print(func(array1 = [4, 1, 2, 1, 1, 2], array2 = [3, 6, 3, 3])) ``