Submission #955174

#	Time	Username	Problem	Language	Result	Execution time	Memory
955174		amine_aroua	Railway Trip 2 (JOI22_ho_t4)	C++17	8 / 100	1205 ms	524288 KiB

Main

#include <bits/stdc++.h>
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
using namespace std;
#define int long long
#define vi vector<int>
#define vl vector<long long>
#define vii vector<pair<int,int>>
#define vll vector<pair<long long,long long>>
#define pb push_back
#define ll long long
#define ld long double
#define nl '\n'
#define boost ios::sync_with_stdio(false)
#define mp make_pair
#define se second
#define fi first
#define fore(i, y) for(int i = 0; i < y; i++)
#define forr(i,x,y) for(int i = x;i<=y;i++)
#define forn(i,y,x) for(int i = y; i >= x; i--)
#define all(v) v.begin(),v.end()
#define sz(v) (int)v.size()
#define clr(v,k) memset(v,k,sizeof(v))
#define rall(v) v.rbegin() , v.rend()
#define pii pair<int,int>
#define pll pair<ll , ll>

const ll MOD = 1e9 + 7;
const ll INF = 1e18 + 1;

ll gcd(ll a , ll b) {return b ? gcd(b , a % b) : a ;} // greatest common divisor (gcd)
ll lcm(ll a , ll b) {return a * (b / gcd(a , b));} // least common multiple (lcm)

// HERE IS THE SOLUTION

signed main()
{
    boost;
    cin.tie(0);
    cout.tie(0);
    int N , K , M;
    cin>>N>>K>>M;
    vector<vi> adj(N + 1);
    fore(i , M)
    {
        int A , B;
        cin>>A>>B;
        if(A < B)
        {
            forr(j , A , min(B-1 , A + K - 1))
            {
                forr(k , j + 1 , B)
                {
                    adj[j].pb(k);
                }
            }
        }
        else
        {
            forn(j , A , max(B + 1 , A - K + 1))
            {
                forn(k , j - 1 , B)
                {
                    adj[j].pb(k);
                }
            }
        }
    }
    int Q;
    cin>>Q;
    while(Q--)
    {
        int S , T;
        cin>>S>>T;
        queue<int> q;
        q.push(S);
        vi dist(N + 1 , INF);
        vector<bool> vis(N + 1 , 0);
        dist[S] = 0;
        while(!q.empty())
        {
            int cur = q.front();
            q.pop();
            if(vis[cur])
                continue;
            vis[cur] = 1;
            if(cur == T)
                break;
            for(auto u : adj[cur])
            {
                if(dist[u] > dist[cur] + 1)
                {
                    dist[u] = dist[cur] + 1;
                    q.push(u);
                }
            }
        }
        cout<<(dist[T] == INF ? -1 : dist[T])<<nl;
    }
}

• Subtask 1: 8 / 8
• Subtask 2: 0 / 8
• Subtask 3: 0 / 11
• Subtask 4: 0 / 25
• Subtask 5: 0 / 35
• Subtask 6: 0 / 13