結果

問題 No.1028 闇討ち
ユーザー imoni_kawaii
提出日時 2020-06-09 23:54:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 1,364 bytes
コンパイル時間 2,146 ms
コンパイル使用メモリ 200,552 KB
最終ジャッジ日時 2025-01-11 00:35:35
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define REP(i, m, n) for(int i = (int)(m); i < (int)(n); ++i)
#define rep(i, n) REP(i, 0, n)
using namespace std;
using ll = long long;
using ld = long double;
template<class T, class U> inline bool chmax(T &a, const U &b) { if(a < b) { a = b; return true; } return false; }
template<class T, class U> inline bool chmin(T &a, const U &b) { if(a > b) { a = b; return true; } return false; }

using pint = pair<int, int>;
int n, a[1000][1000];

int main() {
  cin.tie(0); ios::sync_with_stdio(false);
  cin >> n;
  vector<vector<pint>> pos(n);
  rep(i, n) rep(j, n) cin >> a[i][j], a[i][j]--;
  rep(i, n) {
    rep(j, n) {
      pos[a[i][j]].push_back({i, j});
    }
  }
  ll ans = 0;
  rep(i, n) {
    vector<ll> cd(n+1, 0), cu(n+1, 0);
    rep(j, n) {
      int l = max(0, pos[i][j].first-pos[i][j].second);
      int r = min(n-1, pos[i][j].first+pos[i][j].second);
      cd[r+1]++;
      cu[l]++;
    }
    vector<ll> down(n+1, 0), up(n+1, 0);
    rep(j, n) {
      cd[j+1] += cd[j];
      down[j+1] += down[j]+cd[j+1];
    }
    for(int j = n-1; j >= 0; j--) {
      cu[j] += cu[j+1];
      up[j] += up[j+1]+cu[j+1];
    }
    vector<ll> s(n);
    rep(j, n) s[j] = down[j]+up[j];
    ll d = (1LL<<60);
    rep(j, n) {
      chmin(d, s[j]);
    }
    rep(j, n) d += pos[i][j].second;
    ans += d;
  }
  cout << ans << '\n';
  return 0;
}
0