PY
py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
def zigzagSquare(n, array):
ctr = 1
for i in range(1, n+1):
for j in range(i):
array[j][n-i+j] = ctr
ctr += 1
for i in range(1, n+1):
for j in range(n-i):
array[i+j][j] = ctr
ctr += 1
n = input()
try:
intn = int(n)
array = [[0]*intn for i in range(intn)]
zigzagSquare(intn, array)
for k in range(intn):
print(" ".join(map(str, array[k])))
except(ValueError, TypeError):
print("Enter a positive number")
Enter to Rename, Shift+Enter to Preview
OUTPUT
Run