P14360 [GeeksforGeeks] Print matrix in diagonal pattern
普及-
题目描述

Given a matrix M of n*n size, the task is to complete the function which prints its elements in a diagonal pattern as depicted below.

Example 1:
Input:
N = 3
mat[][] = {{1 2 3},{4 5 6},{7 8 9}}
Output: 1 2 4 7 5 3 6 8 9
Example 2:
Input:
N = 2
mat[][] = {{1 2},{3 4}}
Output: 1 2 3 4
Your Task:
You only need to implement the given function matrixDiagonally() which returns a list containing the matrix diagonally. Do not read input, instead use the arguments given in the function. Print the elements in Matrix in a diagonal pattern.
Expected Time Complexity: O(N*M)
Expected Auxiliary Space: O(1)
Constraints:
1<=N<=100

输入格式

见题面

输出格式

见题面

样例输入
N = 3 mat[][] = {{1 2 3},{4 5 6},{7 8 9}}
样例输出
1 2 4 7 5 3 6 8 9