-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathb.cpp
More file actions
54 lines (48 loc) · 1.37 KB
/
Copy pathb.cpp
File metadata and controls
54 lines (48 loc) · 1.37 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using vll = vector<ll>;
using vvll = vector<vll>;
using vvvll = vector<vvll>;
#define REP(i, n, m) for(ll i=n; i<(ll)m; ++i)
#define IREP(i, n, m) for(ll i=n-1; i>=m; --i)
#define rep(i, n) REP(i, 0, n)
#define irep(i, n) IREP(i, n, 0)
#define all(v) v.begin(), v.end()
#define vprint(v) for(auto e:v){cout<<e<<" ";};cout<<endl;
#define vvprint(vv) for(auto v:vv)vprint(v);
int main(){
cin.tie(0);
ios::sync_with_stdio(false);
cout << setprecision(20);
ll N;
cin >> N;
vll P(N*N, 0);
rep(i, N*N) cin >> P[i];
vll d(N*N, 0), s(N*N, 0);
rep(y, N) rep(x, N) d[y*N+x] = min({x, N-x-1, y, N-y-1});
vvll dirs = vvll{{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
ll total = 0;
rep(i, N*N){
ll p = P[i] - 1;
s[p] = 1;
queue<ll> q;
q.push(p);
total += d[p];
while(!q.empty()){
ll xy = q.front();
ll x = xy%N, y = xy/N;
q.pop();
for(auto &dir : dirs){
ll nx = x+dir[0], ny = y+dir[1];
ll nxy = ny*N + nx;
if(nx<0 || nx>=N || ny<0 || ny>=N) continue;
if(d[nxy]>d[xy]+1 || (s[xy] && d[nxy]>d[xy])){
--d[nxy];
q.push(nxy);
}
}
}
}
cout << total << endl;
}