import random from itertools import * from math import factorial alph = 'pyfgcrlaoeuidhtnsqjkxbmwvz' L = 200 K = 5 combos = [''.join(x) for x in combinations(alph, r=K)] with open('test/prefix/mytest.in', 'w') as f: f.write(f'{len(combos)} {K}\n') for w in combos: f.write(w) f.write('\n') # f.write(''.join(random.choice(combos) for _ in range(10))) f.write(''.join(combos[:4]))