Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def check(a):
- prime = {2, 3, 5, 7, 11, 13, 17, 19, 23}
- for i in range(4):
- s1 = 0
- s2 = 0
- for j in range(4):
- s1 += a[i][j]
- s2 += a[j][i]
- if not (s1 in prime and s2 in prime):
- return 0
- return 1
- def gen(a, n):
- global ans
- if n == 16:
- ans += check(a)
- return
- x = n // 4
- y = n % 4
- for p in {1, 4}:
- a[x][y] = p
- gen(a, n + 1)
- ans = 0
- a = [[-1000 for _ in range(4)] for _ in range(4)]
- gen(a, 0)
- print(ans)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement