Submission #960672

#	Time	Username	Problem	Language	Result	Execution time	Memory
960672		3omar_ahmed	Jakarta Skyscrapers (APIO15_skyscraper)	C++17	0 / 100	1 ms	348 KiB

skyscraper

#include <bits/stdc++.h>
using namespace std ;
#define int long long
#define endl '\n'
#define all(a) a.begin() , a.end()
#define alr(a) a.rbegin() , a.rend()

signed main(){
    ios_base::sync_with_stdio(0); cin.tie(0);cout.tie(0);

    int n, m;
    cin >> n >> m;
    pair < int , int > first, second;
    vector < vector < int >> arr(1 + n);
    for(int i = 0 ; i < m ; i++) {
        int a, b;
        cin >> a >> b;
        if(i == 0) {
            first = {a, b};
        } else if(i == 1) {
            second = {a, b};
        }
        arr[a].push_back(b);
    }
    int nodes = ((n + 1) * (n + 1)) + 5;
    vector < vector < pair < int , int >>> adj(nodes);
    for(int i = 0 ; i < n ; i++) {
        for(int j = 1 ; j <= n ; j++) {
            int node = i * (n + 1) + j;
            if(i + j < n) {
                // cout << node << " " << (i + j) * (n + 1) + j << " 1" << endl;
                adj[node].push_back({(i + j) * (n + 1) + j, 1});
                // cout << node << " " << (i + j) * (n + 1) + n + 1 << " 1" << endl;
                adj[node].push_back({(i + j) * (n + 1) + n + 1, 1});
            }
            if(i - j >= 0) {
                // cout << node << " " << (i - j) * (n + 1) + j << " 1" << endl;
                adj[node].push_back({(i - j) * (n + 1) + j, 1});
                // cout << node << " " << (i - j) * (n + 1) + n + 1 << " 1" << endl;
                adj[node].push_back({(i - j) * (n + 1) + n + 1, 1});
            }
        }
        for(auto j : arr[i]) {
            // cout << (i * (n + 1)) + n + 1 << " " << (i * (n + 1)) + j << " " << 0 << endl;
            adj[i * (n + 1) + n + 1].push_back({i * (n + 1) + j, 0});
        }
    }
    int st = first.first * (n + 1) + first.second, ed = second.first * (n + 1) + second.second; 
    vector < int > dis(nodes, 1e9);
    priority_queue < pair < int , int >> pq;
    pq.push({0, st});
    dis[st] = 0; 
    while(pq.size()) {
        auto top = pq.top();
        pq.pop();
        // if(dis[top.second] != -top.first)    
        //     continue;
        for(auto ch : adj[top.second]) {
            if(dis[ch.first] > dis[top.second] + ch.second) {
                dis[ch.first] = dis[top.second] + ch.second;
                pq.push({-dis[ch.second], ch.first});
            }
        }
    }
    cout << dis[ed] << endl;
    return 0 ;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 12
• Subtask 3: 0 / 14
• Subtask 4: 0 / 21
• Subtask 5: 0 / 43