結果

問題 No.1028 闇討ち
ユーザー SSRSSSRS
提出日時 2020-04-17 22:10:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 221 ms / 2,000 ms
コード長 1,249 bytes
コンパイル時間 1,554 ms
コンパイル使用メモリ 176,964 KB
実行使用メモリ 16,000 KB
最終ジャッジ日時 2024-10-03 13:20:49
合計ジャッジ時間 4,629 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
  int N;
  cin >> N;
  vector<vector<int>> A(N, vector<int>(N));
  for (int i = 0; i < N; i++){
    for (int j = 0; j < N; j++){
      cin >> A[i][j];
      A[i][j]--;
    }
  }
  vector<vector<pair<int, int>>> P(N);
  for (int i = 0; i < N; i++){
    for (int j = 0; j < N; j++){
      P[A[i][j]].push_back(make_pair(i, j));
    }
  }
  long long total = 0;
  for (int i = 0; i < N; i++){
    //社員iに攻撃
    //1行目のコスト
    int row1 = 0;
    for (int j = 0; j < N; j++){
      row1 += max(P[i][j].first, P[i][j].second);
    }
    //1次imos
    vector<int> d(N + 1, 0);
    for (int j = 0; j < N; j++){
      int r = P[i][j].first;
      int c = P[i][j].second;
      if (r - c > 0){
        //減少
        d[1]--;
        if (r - c + 1 <= N){
          d[r - c + 1]++;
        }
      }
      if (r + c < N - 1){
        //増加
        d[r + c + 1]++;
        d[N]--;
      }
    }
    //累積和
    for (int j = 0; j < N; j++){
      d[j + 1] += d[j];
    }
    for (int j = 0; j < N; j++){
      d[j + 1] += d[j];
    }
    int ans = row1;
    for (int j = 0; j < N; j++){
      ans = min(ans, row1 + d[j]);
    }
    total += ans;
  }
  cout << total << endl;
}
0