def main1(n,a): iary=[[] for _ in range(n)] ans=0 for i in range(n): for j in range(n): iary[a[i][j]-1].append([i,j]) for v in range(n): ary=iary[v] mat=[0]*n tmp=0 for x,y in ary: # ロスのない場所 x-y,x+y tmp+=y if x-y>0: cnt=1 for z in range(x-y-1,-1,-1): mat[z]+=cnt cnt+=1 if x+y+1