Distinct sums

def square(n):
    if n==2:
        return [[1,2],[0,0]]
    if n%2 == 1:
        return False
    sub = square(n // 2)
    if sub == False:
        return False
    arr = [[0 for i in range(n)] for i in range(n)]
    for i in range(n//2):
        arr[i][:n//2] = sub[i]
        arr[i+n//2][n//2:] = sub[i]
        arr[i+n//2][:n//2] = [2] * (n // 2)
    return arr

n = int(input())
arr = square(n)
if arr:
    print("YES")
    print("\n".join(" ".join(str(x) for x in row) for row in arr))
else:
    print("IMPOSSIBLE")
#for i in range(n):
#    print(sum(arr[i][j] for j in range(n)))
#    print(sum(arr[j][i] for j in range(n)))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: ACCEPTED

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: ACCEPTED

Test 12

Verdict: WRONG ANSWER

Test 13

Verdict: ACCEPTED