ahelwer / computerized-mathematics Goto Github PK

from itertools import product

N = 4

def sum_other(c, p):
	a = 0
	for o in range(1, N):
		a += c[(p + o) % N]
	return a % N

def problem(c):
	for p in range(N):
		a = sum_other(c, p)
		#d = (a + p) % N
		d = (N - a + p) % N
		if d == c[p]:
			return True
	return False

for c in product(range(N), repeat = N):
	if not problem(c):
		print(c)
		quit()

print("Complete")