Answer to Question #262022 in Python for prime

Question

For this program, I want an exact output with all test cases passed

M, N = input('Enter M, N: ').split()

M, N = int(M), int(N)

matrix = []

for _ in range(M):

row = [int(x) for x in input().split()]

matrix.append(row)

K = int(input('Enter K: '))

values = matrix[0][:-1] + [x[-1] for x in matrix][:-1] + matrix[-1][::-1][:-1] + [x[0] for x in matrix][::-1][:-1]

values = values[-K:] + values[:-K]

output = matrix

idxs = [(0, j) for j in range(N)][:-1] + [(i, N - 1) for i in range(M)][:-1] + [(M - 1, j) for j in range(N)][::-1][:-1] + [(i, 0) for i in range(M)][::-1][:-1]

idx = 0

for i, j in idxs:

output[i][j] = values[idx]

idx += 1

for i in output:

for j in i:

print(j, end=' ')

print()

Sample Input 1

4 4

1 2 3 4

5 6 7 8

9 10 11 12

13 14 15 16

3

Sample Output 1

13 9 5 1

14 7 11 2

15 6 10 3

16 12 8 4

Sample Input 2

3 4

1 2 3 4

10 11 12 5

9 8 7 6

2

Sample Output 2

9 10 1 2

8 11 12 3

7 6 5 4

Answer 1

'''The code was written wrong, What i did here is moved some codes where they should, 

so now it is working for all examples(samples) you have shared. After using for-loop you need to have 4 spaced in the next line'''

M, N = input('Enter M, N: ').split()

M, N = int(M), int(N)

matrix = []

for _ in range(M):

  row = [int(x) for x in input().split()]

  matrix.append(row)

K = int(input('Enter K: '))

values = matrix[0][:-1] + [x[-1] for x in matrix][:-1] + matrix[-1][::-1][:-1] + [x[0] for x in matrix][::-1][:-1]

values = values[-K:] + values[:-K]

output = matrix

idxs = [(0, j) for j in range(N)][:-1] + [(i, N - 1) for i in range(M)][:-1] + [(M - 1, j) for j in range(N)][::-1][:-1] + [(i, 0) for i in range(M)][::-1][:-1]

idx = 0

for i, j in idxs:

  output[i][j] = values[idx]

  idx += 1

for i in output:

  for j in i:

    print(j, end=' ')

  print()