結果

問題 No.1028 闇討ち
ユーザー chocopuuchocopuu
提出日時 2020-04-19 14:30:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 308 ms / 2,000 ms
コード長 1,417 bytes
コンパイル時間 2,364 ms
コンパイル使用メモリ 204,628 KB
最終ジャッジ日時 2025-01-09 21:33:28
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include "bits/stdc++.h"
using namespace std;
#define int long long
#define REP(i, n) for (int i = 0; i < (int)n; ++i)
#define RREP(i, n) for (int i = (int)n - 1; i >= 0; --i)
#define FOR(i, s, n) for (int i = s; i < (int)n; ++i)
#define RFOR(i, s, n) for (int i = (int)n - 1; i >= s; --i)
#define ALL(a) a.begin(), a.end()
#define IN(a, x, b) (a <= x && x < b)
template<class T>inline void out(T t){cout << t << "\n";}
template<class T,class... Ts>inline void out(T t,Ts... ts){cout << t << " ";out(ts...);}
template<class T>inline bool CHMIN(T&a,T b){if(a > b){a = b;return true;}return false;}
template<class T>inline bool CHMAX(T&a,T b){if(a < b){a = b;return true;}return false;}
constexpr int INF = 1e18;



signed main(){
	int N;
	cin >> N;
	vector<vector<int>>a(N,vector<int>(N));
	vector<vector<pair<int,int>>>pos(N);
	REP(i,N){
		REP(j,N){
			cin >> a[i][j];
			--a[i][j];
			pos[a[i][j]].emplace_back(i,j);
		}
	}
	int ans = 0;
	REP(i,N){
		vector<int>imos1(N + 1),imos2(N + 1);
		for(auto[x,y]:pos[i]){
			++imos1[min(N,x + y + 1)];
			--imos1[N];
			++imos2[0];
			--imos2[max(0ll,x - y)];
		}
		REP(j,N){
			imos1[j + 1] += imos1[j];
			imos2[j + 1] += imos2[j];
		}
		REP(j,N)imos1[j + 1] += imos1[j];
		RREP(j,N)imos2[j] += imos2[j + 1];
		int mi = INF,idx = -1;
		REP(j,N)if(CHMIN(mi,imos1[j] + imos2[j]))idx = j;
		REP(j,N){
			auto[x,y] = pos[i][j];
			ans += max(abs(idx-x),y);
		}
	}
	out(ans);
}
0