結果

問題 No.1028 闇討ち
ユーザー Dente
提出日時 2020-04-18 19:37:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 246 ms / 2,000 ms
コード長 1,356 bytes
コンパイル時間 1,770 ms
コンパイル使用メモリ 173,672 KB
実行使用メモリ 16,176 KB
最終ジャッジ日時 2024-10-04 00:13:56
合計ジャッジ時間 5,222 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < (n); i++)
#define ALL(v) (v).begin(), (v).end()
using ll = long long;
using P = pair<int, int>;
constexpr int INF = 1e9;
constexpr long long LINF = 1e18;
constexpr long long MOD = 1e9 + 7;

signed main() {
    int n;
    cin >> n;
    vector<P> pos[n];
    int a[n][n];
    rep(i, n) {
        rep(j, n) {
            cin >> a[i][j];
            a[i][j]--;
            pos[a[i][j]].emplace_back(i, j);
        }
    }
    ll ans = 0;
    rep(i, n) {
        ll imos0[n + 1] = {}, imos1[n + 1] = {}, imos1r[n + 1] = {};
        rep(j, n) {
            int y = pos[i][j].first, x = pos[i][j].second;
            imos0[0] += x;
            if (y + x + 1 <= n) imos1[y + (x + 1)]++;
            if (y - (x + 1) >= 0) imos1r[y - (x + 1)]++;
        }
        rep(j, n) {
            imos0[j + 1] += imos0[j];
            imos1[j + 1] += imos1[j];
        }
        for (int j = n; j > 0; j--) {
            imos1r[j - 1] += imos1r[j];
        }
        rep(j, n) {
            imos1[j + 1] += imos1[j];
        }
        for (int j = n; j > 0; j--) {
            imos1r[j - 1] += imos1r[j];
        }
        ll mn = INF;
        rep(j, n) {;
            mn = min(mn, imos0[j] + imos1[j] + imos1r[j]);
        }
        ans += mn;
    }
    cout << ans << endl;
    return 0;
}
0