結果

問題 No.1028 闇討ち
ユーザー RhoRho
提出日時 2020-03-28 19:34:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 274 ms / 2,000 ms
コード長 1,060 bytes
コンパイル時間 1,764 ms
コンパイル使用メモリ 173,684 KB
実行使用メモリ 28,896 KB
最終ジャッジ日時 2024-07-03 00:30:38
合計ジャッジ時間 5,559 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:51:39: warning: 'pos2' may be used uninitialized [-Wmaybe-uninitialized]
   51 |                         ans += max(abs(x - pos2), y);
      |                                    ~~~^~~~~~~~~~
main.cpp:42:21: note: 'pos2' was declared here
   42 |                 int pos2, mx = 1e17;
      |                     ^~~~

ソースコード

diff # 

#include "bits/stdc++.h"
#pragma warning(disable:4996)
using namespace std;
#define int long long
#define rep(i,n) for(int i=0;i<n;i++)
typedef pair<int, int> P;
int a[1005][1005];
vector<P>pos[1005];

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

	rep(i, n) {
		rep(j, n) {
			cin >> a[i][j];
			a[i][j]--;
			pos[a[i][j]].push_back(P(i, j));
		}
	}

	int ans = 0;

	rep(I, n) {
		vector<int>imos(n + 1);
		vector<int>imos1(n + 1), imos2(n + 1);
		assert(pos[I].size() == n);
		rep(j, n) {
			int x = pos[I][j].first, y = pos[I][j].second;
			imos1[0]++;
			imos1[max(x - y,0ll)]--;
			imos2[min(x + y + 1,n)]++;
			imos2[n]--;
		}
		rep(i, n) {
			imos1[i + 1] += imos1[i];
			imos2[i + 1] += imos2[i];
		}
		for (int i = n; i > 0; i--)imos1[i - 1] += imos1[i];
		rep(i, n)imos2[i + 1] += imos2[i];
		rep(i, n)imos[i] = imos1[i] + imos2[i];
		
		int pos2, mx = 1e17;
		rep(i, n) {
			if (mx > imos[i]) {
				mx = imos[i]; pos2 = i;
			}
		}

		rep(j, n) {
			int x = pos[I][j].first, y = pos[I][j].second;
			ans += max(abs(x - pos2), y);
		}
	}
	cout << ans << endl;
}
0