Print Diagonal

Moderate

0/80

Average time to solve is 30m

Contributed by

3 upvotes

Problem statement

You are given a 2D matrix, your task is to return a 2D vector containing all elements of the matrix in a diagonal fashion.

Example:

Following will be the output of the above matrix:

Detailed explanation ( Input/output format, Notes, Images )

Input Format:

The first line contains an Integer 'T' which denotes the number of test cases or queries to be run. Then the test cases follow.

The first line of each test case contains two space-separated integers ‘N’ and ‘M’ denoting the number of rows and columns of the matrix respectively.

N’ lines follow. Each of the next ‘N’ lines contains ‘M’ space-separated integers separated by space.

Output Format:

For each test case, return a 2D vector containing all elements of the matrix in a diagonal fashion.

The output of each test case should be printed in a separate line.

Note:

You are not required to print anything, it has already been taken care of. Just implement the function.

Constraints :

1 <= T <= 10
1 <= N <= 100
1 <= M <= 100
1 <= mat[i][j] <= 100

Time Limit : 1 sec.

Sample Input 1:

2
4 6
1 2 3 4 5 6
7 8 9 10 11 12 
13 14 15 16 17 18
19 20 21 22 23 24
4 4
1 2 3 4
6 7 8 9
11 12 13 14
16 17 18 19

Sample Output 1:

Explanation For Sample Output 1 :

Test Case 1:

In the above pic, arrow lines represent the diagonals of the matrix which are to be returned.
Test Case 2:

In the above pic, arrow lines represent the diagonals of the matrix which are to be returned.

Sample Input 2:

2
5 4
1 2 3 4
5 6 7 8
9 10 11 12 
13 14 15 16
17 18 19 20
2 2
8 3
6 1

Sample Output 2:

Console