Answer to Question #224329 in Python for Sai

Question

Zig-zag order in matrix

Input:---
4 4
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16

Output :--
1 2 3 4 5 6 7 8 9 10 11 12 16 15 14 13

Input 2:---

3 4
1 2 3 4
10 11 12 5
9 8 7 6

Output 2:--

1 2 3 4 5 12 11 10 9 8 7 6

Answer 1

def read_matrix():
    line = input()
    words =line.split()
    n = int(words[0])
    m = int(words[1])
    mat = []
    for i in range(n):
        row = []
        line = input()
        words = line.split()
        for j in range(m):
            row.append(int(words[j]))
        mat.append(row)
    return mat


def print_zig_zag(mat):
    n = len(mat)
    m = len(mat[0])
    for i in range(n):
        for j in range(m):
            jj = j if i%2==0 else m-1-j
            print(mat[i][jj], end=' ')
    print()


def main():
    mat = read_matrix()
    print_zig_zag(mat)


if __name__ == '__main__':
    main()