結果
| 問題 | 
                            No.1028 闇討ち
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2021-06-18 17:48:55 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                AC
                                 
                             
                            
                         | 
                    
| 実行時間 | 1,317 ms / 2,000 ms | 
| コード長 | 682 bytes | 
| コンパイル時間 | 206 ms | 
| コンパイル使用メモリ | 82,280 KB | 
| 実行使用メモリ | 201,644 KB | 
| 最終ジャッジ日時 | 2024-06-22 14:31:21 | 
| 合計ジャッジ時間 | 13,499 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge3 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 | 
| other | AC * 20 | 
ソースコード
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<n:
        cnt=1
        for z in range(x+y+1,n):
          mat[z]+=cnt
          cnt+=1
    tmp+=min(mat)
    ans+=tmp
    #print(v,":",tmp,mat)
  return ans
if __name__=='__main__':
  n=int(input())
  a=[list(map(int,input().split())) for _ in range(n)]
  ret1=main1(n,a)
  print(ret1)