結果

問題 No.1028 闇討ち
コンテスト
ユーザー leaf_1415
提出日時 2020-04-17 21:54:57
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 526 ms / 2,000 ms
コード長 2,223 bytes
コンパイル時間 918 ms
コンパイル使用メモリ 83,804 KB
実行使用メモリ 71,236 KB
最終ジャッジ日時 2024-10-03 12:44:19
合計ジャッジ時間 6,698 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#define llint long long
#define inf 1e18
#define rep(x, s, t) for(llint (x) = (s); (x) < (t); (x)++)
#define Rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))

using namespace std;
typedef pair<llint, llint> P;

llint n;
llint a[1005][1005];
vector<llint> vec[1005][1005];
llint d[1005][1005];
llint s[1005][1005];

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n;
	for(int y = 1; y <= n; y++){
		for(int x = 1; x <= n; x++){
			cin >> a[x][y];
		}
	}
	
	
	for(int y = 1; y <= n; y++){
		for(int x = 1; x <= n; x++){
			if(y+x-1 <= n) vec[a[x][y]][y+x-1].push_back(x-1);
		}
	}
	for(int i = 1; i <= n; i++){
		llint sum = 0, cnt = 0;
		for(int y = 1; y <= n; y++){
			s[i][y] += sum;
			for(int j = 0; j < vec[i][y].size(); j++){
				llint x = vec[i][y][j];
				sum += x;
				cnt++;
			}
			sum += cnt;
		}
	}
	
	
	for(int y = 1; y <= n; y++){
		for(int x = 1; x <= n; x++){
			vec[x][y].clear();
		}
	}
	for(int y = 1; y <= n; y++){
		for(int x = 1; x <= n; x++){
			if(y-(x-1) >= 1) vec[a[x][y]][y-(x-1)].push_back(x-1);
		}
	}
	for(int i = 1; i <= n; i++){
		llint sum = 0, cnt = 0;
		for(int y = n; y >= 1; y--){
			s[i][y] += sum;
			for(int j = 0; j < vec[i][y].size(); j++){
				llint x = vec[i][y][j];
				sum += x;
				cnt++;
			}
			sum += cnt;
		}
	}
	
	
	for(llint y = 1; y <= n; y++){
		for(llint x = 1; x <= n; x++){
			llint id = a[x][y];
			llint l = max(1LL, y-(x-1)), r = min(n, y+(x-1));
			if(l > r) continue;
			d[id][l] += x-1, d[id][r+1] -= x-1;
		}
	}
	for(int i = 1; i <= n; i++){
		for(int y = 1; y <= n; y++) d[i][y] += d[i][y-1];
	}
	
	for(int i = 1; i <= n; i++){
		for(int y = 1; y <= n; y++) s[i][y] += d[i][y];
	}
	
	
	llint ans = 0;
	for(int i = 1; i <= n; i++){
		llint mn = inf;
		for(int y = 1; y <= n; y++) mn = min(mn, s[i][y]);
		ans += mn;
	}
	cout << ans << endl;
	
	return 0;
}
0