Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- m = int(input())
- arr = [ int(k) for k in input().split() ]
- mat = []
- k=0
- #converting into m x m matrix
- for i in range(m):
- temp =[]
- for j in range(m):
- temp.append(arr[k])
- k+=1
- mat.append(temp)
- arr = mat
- for i in range(m):
- if i==0 :
- arr[i][i] = arr[i][i+1] + arr[i+1][i] + arr[i+1][i+1]
- elif i==m-1 :
- arr[i][i] = arr[i][i-1] + arr[i-1][i-1] + arr[i-1][i]
- else :
- 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]
- print(arr)
Advertisement
Add Comment
Please, Sign In to add comment