0

The program must accept an integer matrix of size RxC and an integer K as the input.The program must print all the integers in the bottom left to top right diagonal in each KxK sub matrix without any overlapping of the given matrix

NOTE: The values of R and C ara always multiples of K

Example :

Input:
4 4
1 2 3 4
5 6 7 9
8 5 4 1
6 9 0 5
2
Output:
5 2
7 4
6 5
0 1

My Code :

row , col = map(int,input().split())
mat = [list(map(int,input().split())) for i in range(row)]
k = int(input())
temp = k
# To store the right diagonal elements of all the sub matrices 
right_diagonal_elements = []
for i in range(row):
    te =temp
    # Iterating through the matrix in such a way
    #that it appends only the k th element ,
    #then k-1, and so on til the first element
    while te<=col:
        # It appends the right diagonal element but in a different order
        right_diagonal_elements.append(mat[i][te-1])
        te+=k
    if temp == 1:
        temp = k
        continue
    temp -= 1

In this my code gets all the right diagonal elements, but I am stuck on how to arrange it in such a way that it obtains the given output
The right_diagonal_elements has [2, 4, 5, 7, 5, 1, 6, 0] for the example above. I am stuck on how to arrange this elements in the list in order to get the output in example

DaVinci
  • 868
  • 1
  • 7
  • 25

1 Answers1

0

If you want to print all the sub matrices right diagonal then 

R, C = map(int, input().split())
mat = [list(map(int, input().split())) for _ in range(R)]
K = int(input())
#iterating from k-1 to R by K
for i in range(K - 1, R, K):
    # Iterating j from 0 to C by K
    for j in range(0, C, K):
        # Iterating through the sub matrix of size KxK
        for p in range(K):

            print(mat[i-p][j+p], end=' ')
        print()
DaVinci
  • 868
  • 1
  • 7
  • 25