답안 #241519

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
241519 2020-06-24T10:57:03 Z osaaateiasavtnl Railway Trip (JOI17_railway_trip) C++17
25 / 100
2000 ms 10476 KB
#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;
}
# 결과 실행 시간 메모리 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 -
# 결과 실행 시간 메모리 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 -
# 결과 실행 시간 메모리 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
# 결과 실행 시간 메모리 Grader output
1 Execution timed out 2082 ms 10476 KB Time limit exceeded
2 Halted 0 ms 0 KB -