Find nth element of spiral matrix

• Problem Description

  
  Given a matrix your task is to find the kth element which is obtained while traversing the matrix spirally. You need to complete the method findK which takes four arguments the first argument is the matrix A and the next two arguments will be n and m denoting the size of the matrix A and then the forth argument is an integer k denoting the kth element . The function will return the kth element obtained while traversing the matrix spirally.
  
  
  Input:
  The first line of input is the no of test cases then T test cases follow . The first line of each test case has three integers n,m and k . Then in the next line are n*m space separated values of the matrix A [ ] [ ] .
  
  Output:
  The output for each test case will be the kth obtained element .
  
  Constraints:
  1<=T<=100
  1<=n,m<=20
  1<=k<=n*m

• CODING ARENA

• a=int(input())

  x=(input().split())

  z=int(x[-1])

  y=input().split()

  print(y[-z])

• 
  

• Test Case 1

  
  Input (stdin)

  1
  
  3 3 4
  
  1 2 3 4 5 6 7 8 9 
  
  

  Expected Output

  6

• Test Case 2

  
  Input (stdin)

  0
  
  

  Expected Output

  0