#include<bits/stdc++.h>
using namespace std;
#define ii pair <int, int>
#define app push_back
#define all(a) a.begin(), a.end()
#define bp __builtin_popcountll
#define ll long long
#define mp make_pair
#define f first
#define s second
#define Time (double)clock()/CLOCKS_PER_SEC
const int N = 1e5+7, INF = 1e9+7;
int n, k, q;
int a[N], ans[N];
int pref[N], suff[N];
int newl[N], newr[N];
vector <int> pos[N];
int par[N];
int root(int u) {
if (par[u] == u)
return u;
else
return par[u] = root(par[u]);
}
void merge(int u, int v) {
if (u == 0 || v == 0)
return;
if (u > n || v > n)
return;
u = root(u);
v = root(v);
if (u != v) {
par[u] = v;
}
}
int link_l[N], link_r[N];
int cnt[N];
signed main() {
#ifdef HOME
freopen("input.txt", "r", stdin);
#else
#define endl '\n'
ios_base::sync_with_stdio(0); cin.tie(0);
#endif
cin >> n >> k >> q;
for (int i = 1; i <= n; ++i)
cin >> a[i];
vector <ii> d;
for (int i = 0; i < q; ++i) {
int u, v;
cin >> u >> v;
if (v < u)
swap(u, v);
d.app(mp(u, v));
ans[i] = INF;
}
for (int i = 1; i <= k; ++i)
pos[i].app(0);
for (int i = 1; i <= n; ++i)
pos[a[i]].app(i);
for (int i = 1; i <= k; ++i)
pos[i].app(n+1);
for (int i = 1; i <= n; ++i)
par[i] = i;
auto same_dist = [&](int i, int j) {
int p1 = lower_bound(all(pos[a[i]]), i) - pos[a[i]].begin();
int p2 = lower_bound(all(pos[a[j]]), j) - pos[a[j]].begin();
return abs(p1-p2);
};
for (int x = 1; x <= k; ++x) {
for (int i : pos[x]) {
if (i && a[i - 1] <= x) {
merge(i, i - 1);
}
if (i + 1 <= n && a[i + 1] <= x) {
merge(i, i + 1);
}
}
memset(link_l, 0, sizeof link_l);
memset(link_r, 0, sizeof link_r);
for (int i = 1; i <= n; ++i)
newl[i] = newr[i] = INF;
for (int i : pos[x]) {
link_l[i] = link_r[i] = i;
newl[i] = newr[i] = 0;
for (int j = i - 1; j; --j) {
if (a[j] >= a[i])
break;
newr[j] = suff[j]+1;
link_r[j] = i;
}
for (int j = i + 1; j <= n; ++j) {
if (a[j] >= a[i])
break;
newl[j] = pref[j]+1;
link_l[j] = i;
}
}
//pref - dist to nearest prefix maximum
//suff - dist to nearest suffix maximum
//newl - dist to nearest left x
//newr - dist to nearest right x
//link_l = number of the nearest left x
auto getnewl = [&](int u) {
return min(newl[u], newr[u]+1);
};
auto getnewr = [&](int u) {
return min(newl[u]+1, newr[u]);
};
int lf = 0;
for (int i = 1; i <= n; ++i) {
if (a[i] > x) {
lf = i;
continue;
}
auto t = upper_bound(all(pos[x]), lf);
if (*t <= i) {
auto link = prev(upper_bound(all(pos[x]), i));
pref[i] = getnewl(i) + (link - t);
}
else if (link_r[i]) {
pref[i] = min(pref[i], getnewr(i));
}
}
int rf = n+1;
for (int i = n; i; --i) {
if (a[i] > x) {
rf = i;
continue;
}
auto t = prev(upper_bound(all(pos[x]), rf));
if (*t >= i) {
auto link = lower_bound(all(pos[x]), i);
suff[i] = getnewr(i) + (t - link);
}
else if (link_l[i]) {
suff[i] = min(suff[i], getnewl(i));
}
}
for (int i = 0; i < q; ++i) {
int u = d[i].f, v = d[i].s;
if (root(u) == root(v) && cnt[i] < 20) {
++cnt[i];
if (link_l[u] && link_l[v]) {
ans[i] = min(ans[i], getnewl(u) + getnewl(v) + same_dist(link_l[u], link_l[v]));
}
if (link_l[u] && link_r[v]) {
ans[i] = min(ans[i], getnewl(u) + getnewr(v) + same_dist(link_l[u], link_r[v]));
}
if (link_r[u] && link_l[v]) {
ans[i] = min(ans[i], getnewr(u) + getnewl(v) + same_dist(link_r[u], link_l[v]));
}
if (link_r[u] && link_r[v]) {
ans[i] = min(ans[i], getnewr(u) + getnewr(v) + same_dist(link_r[u], link_r[v]));
}
}
}
}
for (int i = 0; i < q; ++i)
cout << ans[i] - 1 << endl;
}
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
9 ms |
3456 KB |
Output is correct |
| 2 |
Correct |
9 ms |
3456 KB |
Output is correct |
| 3 |
Incorrect |
9 ms |
3456 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
39 ms |
3704 KB |
Output is correct |
| 2 |
Correct |
82 ms |
6520 KB |
Output is correct |
| 3 |
Correct |
118 ms |
6528 KB |
Output is correct |
| 4 |
Correct |
195 ms |
6392 KB |
Output is correct |
| 5 |
Correct |
303 ms |
6520 KB |
Output is correct |
| 6 |
Correct |
1860 ms |
6520 KB |
Output is correct |
| 7 |
Execution timed out |
2094 ms |
9340 KB |
Time limit exceeded |
| 8 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
213 ms |
8560 KB |
Output is correct |
| 2 |
Correct |
199 ms |
8824 KB |
Output is correct |
| 3 |
Correct |
219 ms |
8564 KB |
Output is correct |
| 4 |
Correct |
226 ms |
8516 KB |
Output is correct |
| 5 |
Correct |
223 ms |
8560 KB |
Output is correct |
| 6 |
Correct |
238 ms |
8556 KB |
Output is correct |
| 7 |
Correct |
239 ms |
8556 KB |
Output is correct |
| 8 |
Correct |
151 ms |
9196 KB |
Output is correct |
| 9 |
Correct |
191 ms |
10220 KB |
Output is correct |
| 10 |
Correct |
262 ms |
9840 KB |
Output is correct |
| 11 |
Correct |
285 ms |
10476 KB |
Output is correct |
| 12 |
Correct |
278 ms |
9964 KB |
Output is correct |
| 13 |
Correct |
264 ms |
9968 KB |
Output is correct |
| 14 |
Correct |
189 ms |
9072 KB |
Output is correct |
| 15 |
Correct |
213 ms |
8816 KB |
Output is correct |
| 16 |
Correct |
241 ms |
8300 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
2082 ms |
10476 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
