結果
| 問題 | 
                            No.1028 闇討ち
                             | 
                    
| コンテスト | |
| ユーザー | 
                             uni_python
                         | 
                    
| 提出日時 | 2020-07-01 22:14:15 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 1,498 bytes | 
| コンパイル時間 | 344 ms | 
| コンパイル使用メモリ | 82,176 KB | 
| 実行使用メモリ | 101,632 KB | 
| 最終ジャッジ日時 | 2024-09-14 04:59:21 | 
| 合計ジャッジ時間 | 5,465 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 2 | 
| other | AC * 3 WA * 17 | 
ソースコード
import sys
input=sys.stdin.readline
def I(): return int(input())
def MI(): return map(int, input().split())
def LI(): return list(map(int, input().split()))
def main():
    mod=10**9+7
    N=I()
    A=[]
    for i in range(N):
        a=LI()
        A.append(a)
        
    C=[0]*N#社員iの定数部分
    Su=[[0]*N for _ in range(N)]#上に見ていく累積2,社員aのj行目
    Sd=[[0]*N for _ in range(N)]#下に
    
    for i in range(N):
        for j in range(N):
            a=A[i][j]-1
            C[a]+=j
            u=i+j+1
            if u<N:
                Su[a][u]+=1
            d=i-j-1
            if d>=0:
                Sd[a][d]=1
            
    for i in range(N):
        for j in range(1,N):
            Su[i][j]+=Su[i][j-1]
    for i in range(N):
        for j in range(1,N):
            Su[i][j]+=Su[i][j-1]
            
    for i in range(N):
        for j in range(N-2,-1,-1):
            Sd[i][j]+=Sd[i][j+1]
    for i in range(N):
        for j in range(N-2,-1,-1):
            Sd[i][j]+=Sd[i][j+1]
            
    # for i in range(N):
    #     print(Su[i])
        
    # for i in range(N):
    #     print(Sd[i])
            
    anss=[N*N]*N#各社員に対して
    for i in range(N):
        for j in range(N):
            temp=Su[i][j]+Sd[i][j]+C[i]
            # print(i,j,temp,C[i])
            anss[i]=min(anss[i],temp)
    ans=sum(anss)
    print(ans)
    # print(anss)
            
        
            
            
            
    
main()
            
            
            
        
            
uni_python