Вывод двумерного массива змейкой

n=int(input())
m=int(input())
a=[[0 for j in range(m)] for i in range(n)]
i=0
j=0
k=0
count=0
a[0][0]=k
while (count<m*n):
    if (i==0) and (j==0):
        i+=1
        k+=1
        a[i][j]=k
        count+=1
    if (j==0):
        while (i!=0):
            k+=1
            a[i][j]=k
            i-=1
            j+=1
            count+=1
        k+=1
        a[i][j]=k
        count+=1
        j+=1
        count+=1
    for q in range(n):
        print(a[q]) 
    if (i==0):
        while (j!=0):
            k+=1
            a[i][j]=k
            j-=1
            i+=1
            count+=1
        k+=1
        a[i][j]=k
        count+=1
        i+=1
        k+=1
        a[i][j]=k
        count+=1
for q in range(n):
    print(a[q])

#  Число - количество столбцов и строк в матрице
N=11
counter=1
 
#z=[[0 for row in range(0,x)] for col in range(0,y)]
arr=[[0 for row in range(0, N)] for col in range(0, N)]
 
row=0
col=0
arr[row][col]=counter
for i in range(N*2):
    #if i == row+col and i==0:
        #break
    #if row == N-1 and col == N-1:
        #counter=counter+1
        #arr[row][col]=counter
        #break
    if i<N:
        if i > row+col and col < N -1 and row < N-1:
            if row == i-1:
                for x in range(i+1):
                    counter=counter+1
                    arr[i-x][x]=counter
                    row=i-x
                    col=x
 
 
            elif col==i-1:
                for y in range(i+1):
                    counter=counter+1
                    arr[y][i-y]=counter
                    row=y
                    col=i-y
    elif i>=N:
        if i>row+col and col<N and row<N:
            if col == N-1:
                for x in range(N*2-i-1):
                    counter=counter+1
                    arr[i-(N-1)+x][N-1-x]=counter
                    row=i-(N-1)+x
                    col=N-1-x
            elif row == N-1:
                for y in range(N*2-i-1):
                    counter=counter+1
                    arr[N-1-y][i-(N-1)+y]=counter
                    row=N-i-1
                    col=i-(N-1)+y
 
print(*arr, sep='\n')

def snake(n, m):
    matrix = [[0]*m for _ in range(n)]
    val = -1
 
    for k in range(n + m):
        tmp = min(k, max(n-1, m-1))
        for j in range(tmp, -1, -1):
            if k%2:
                row = j
                col = k - j 
            else:
                row = k - j 
                col = j
            
            if row < n and col < m:
                val += 1
                matrix[row][col] = val
 
    return matrix
     
res = snake(5,3) #snake(int(input('rows = ')), int(input('columns = ')))
k = len(str(res[-1][-1]))
for row in res:
    print(' '.join([f'{elem:{k}d}' for elem in row]))