結果
問題 | No.1028 闇討ち |
ユーザー |
|
提出日時 | 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 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define int long longtemplate<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;}