In [15]:
import sys
def main():
    n = int(input().strip())
    a = [int(a_t) for a_i in range(n) for a_t in input().strip().split(' ')]
    d1_sum=0
    d2_sum=0
    
    
    
    
    for i,j in range(n*n):
        
        if i == j:
            
            d1_sum+=a[i]
            d2_sum+=a[i]
        if i in(0,(n*n)-1):
            print("First : {}".format(a[i]))
            d1_sum+=a[i]
        elif i in (n-1,n*2):
            print("Second : {}".format(a[i]))
            d2_sum+=a[i]
    d_diff = d1_sum-d2_sum
    print(abs(d_diff))
if __name__ == '__main__': main()


4
-1 1 -7 -8
-10 -8 -5 -2
0 9 7 -1
4 4 -2 1
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-15-b5df385f9a65> in <module>()
     23     d_diff = d1_sum-d2_sum
     24     print(abs(d_diff))
---> 25 if __name__ == '__main__': main()
     26 

<ipython-input-15-b5df385f9a65> in main()
      9 
     10 
---> 11     for i,j in range(n*n):
     12 
     13         if i == j:

TypeError: 'int' object is not iterable

In [ ]:
currentMax = 0
        result = 0

        for i in range(len(inputArray)):
                currentMax = inputArray[i]
                result = max(result, currentMax)

        return result