th0m45s5helby

Untitled

Apr 28th, 2022
713
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.69 KB | None | 0 0
  1. m = int(input())
  2.  
  3. arr = [ int(k) for k in input().split() ]
  4.  
  5. mat = []
  6.  
  7. k=0
  8.  
  9. #converting into m x m matrix
  10. for i in range(m):
  11.    
  12.     temp =[]
  13.    
  14.     for j in range(m):
  15.        
  16.         temp.append(arr[k])
  17.        
  18.         k+=1
  19.    
  20.     mat.append(temp)
  21.  
  22. arr = mat
  23.  
  24. for i in range(m):
  25.    
  26.  
  27.     if i==0 :
  28.        
  29.         arr[i][i] = arr[i][i+1] + arr[i+1][i] + arr[i+1][i+1]
  30.    
  31.     elif i==m-1 :
  32.        
  33.         arr[i][i] = arr[i][i-1] + arr[i-1][i-1] + arr[i-1][i]
  34.    
  35.     else :
  36.        
  37.         arr[i][i] = arr[i][i-1] + arr[i][i+1] + arr[i-1][i] + arr[i+1][i] + arr[i-1][i-1] + arr[i+1][i+1] + arr[i-1][i+1] + arr[i+1][i-1]
  38.  
  39.  
  40.  
  41. print(arr)
  42.        
  43.  
  44.  
  45.  
Advertisement
Add Comment
Please, Sign In to add comment