結果

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define int long long
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }

template <typename T>
struct cumulative_sum_linear {
    int n;
    vector<T> x, a, b;

    cumulative_sum_linear(int n_ = 0) : n(n_), x(n, 0), a(n + 1, 0), b(n + 1, 0) {}

    void add(int l, int r, T aa, T bb) {
        a[l] += aa;
        a[r] -= aa;
        a[l] -= bb * l;
        a[r] += bb * l;
        b[l] += bb;
        b[r] -= bb;
    }

    void exec() {
        for (int i = 0; i < n; ++i) {
            x[i] = a[i] + b[i] * i;
            a[i + 1] += a[i];
            b[i + 1] += b[i];
        }
    }

    T operator[](int i) const { return x[i]; }
};

signed main(){
	int n;
	cin >> n;

	vector<pair<int, int>> a[1010];
	for(int i = 0;i < n;i++){
		int b[1010];
		for(int j = 0;j < n;j++){
			cin >> b[i];
			a[b[i]-1].push_back({i, j});
		}
	}

	int ans = 0;
	for(int num = 0;num < n;num++){

		cumulative_sum_linear<int> imos(n);
		for(int x = 0;x < n;x++){
			int i = a[num][x].first;
			int d = a[num][x].second;
			int l = max(i-d, 0ll);
			int r = min(i+d, n-1ll);
			if(l-1 >= 0) imos.add(0ll, l, i, -1);
			imos.add(l, r+1, d, 0);
			if(r+1 < n) imos.add(r+1, n, d+1, 1);
		}

		imos.exec();

		int mi = LLONG_MAX;
		for(int i = 0;i < n;i++){
			chmin(mi, imos[i]);
		}

		ans += mi;
	}

	cout << ans << endl;

	return 0;
}
0