Submission #956567

#	Time	Username	Problem	Language	Result	Execution time	Memory
956567		ZHIRDILBILDIZ	Passport (JOI23_passport)	C++14	100 / 100	1686 ms	187596 KiB

passport

#include <bits/stdc++.h>
#define fi first
#define se second
#define ll long long

using namespace std;
const ll N = (1 << 18);
const ll inf = 1e9;

ll n;
ll dist[3 * N + 10];
vector<pair<ll, ll>> gr[3 * N + 10];

void build (ll l = 1, ll r = N, ll v = 1) {
    if (l == r) {
        gr[l].push_back({v + N, 0});
        return;
    }
    ll mid = (l + r) >> 1;
    build (l, mid, v << 1);
    build (mid + 1, r, v << 1 ^ 1);
    gr[(v << 1) + N].push_back({v + N, 0});
    gr[((v << 1) ^ 1) + N].push_back({v + N, 0});
}

void add (ll l1, ll r1, ll city, ll l = 1, ll r = N, ll v = 1) {
    if (l > r1 || r < l1)
        return;
    if (l1 <= l && r <= r1) {
        gr[v + N].push_back({city, 1});
        return;
    }
    ll mid = (l + r) >> 1;
    add (l1, r1, city, l, mid, v << 1);
    add (l1, r1, city, mid + 1, r, v << 1 ^ 1);
}

vector<ll> deikstra (vector<pair<ll, ll>> v) {
    vector<ll> dist(3 * N + 1, inf);
    for (auto i : v)
        dist[i.fi] = i.se;
    set<pair<ll, ll>> st;
    for (ll i = 0; i < dist.size(); ++i)
        st.insert({dist[i], i});
    while (st.size()) {
        auto p = *st.begin();
        st.erase(st.begin());
        for (auto i : gr[p.se])
            if (dist[i.fi] > dist[p.se] + i.se) {
                st.erase({dist[i.fi], i.fi});
                dist[i.fi] = dist[p.se] + i.se;
                st.insert({dist[i.fi], i.fi});
            }
    }
    vector<ll> now(n + 1);
    for (ll i = 1; i <= n; ++i)
        now[i] = dist[i];
    return now;
}

signed main () {
    ios_base::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    build();
    cin >> n;
    for (ll i = 1; i <= n; ++i) {
        ll l, r;
        cin >> l >> r;
        add (l, r, i);
    }

    vector<ll> dist_from_one = deikstra({{1, 0}});
    vector<ll> dist_from_n = deikstra({{n, 0}});

    vector<pair<ll, ll>> v;
    for (ll i = 1; i <= n; ++i)
        v.push_back({i, dist_from_n[i] + dist_from_one[i] - (1 < i && i < n)});
    vector<ll> ans = deikstra(v);

    ll q;
    cin >> q;
    while (q--) {
        ll x;
        cin >> x;
        if (ans[x] >= inf)
            cout << "-1\n";
        else
            cout << ans[x] << '\n';
    }
    return 0;
}

Compilation message (stderr)

passport.cpp: In function 'std::vector<long long int> deikstra(std::vector<std::pair<long long int, long long int> >)':
passport.cpp:43:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |     for (ll i = 0; i < dist.size(); ++i)
      |                    ~~^~~~~~~~~~~~~

• Subtask 1: 6 / 6
• Subtask 2: 16 / 16
• Subtask 3: 24 / 24
• Subtask 4: 8 / 8
• Subtask 5: 46 / 46