New update is available. Click here to update.

Solve problems & track your progress

Checkout your overall progress in every topic here

Become

Sensei

in DSA topics

Open the topic and solve more problems associated with it to improve your skills

Check out the skill meter for every topic

See how many problems you are left with to solve for cracking any stage. Score more than zero to get your progress counted.

Problem

Submissions

Print Diagonal

Contributed by

Archit Sharma

Last Updated: 23 Feb, 2023

Medium

0/80

Avg time to solve 30 mins

Success Rate 80 %

2 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:

Auto

Console