1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
def getSum(rank, a):
    N = len(a)
    
    rst = 0
    for idx, val in enumerate(a):
        # count students
        ## left
        left = 0
        if idx-2<0:
            for left_i in range(0,idx):
                if rank[left_i]<rank[idx]:
                    left += 1
        else:
            for left_i in range(idx-2,idx):
                if rank[left_i]<rank[idx]:
                    left += 1
        ## right
        right = 0
        if idx+2<N:
            for right_i in range(idx+1,idx+3):
                if rank[right_i]<rank[idx]:
                    right += 1
        else:
            for right_i in range(idx+1,N):
                if rank[right_i]<rank[idx]:
                    right += 1
        
        # curr
        rst += a[idx][left+right]
    return rst
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX