Submission #882253

#	Time	Username	Problem	Language	Result	Execution time	Memory
882253		kokoue	Evacuation plan (IZhO18_plan)	C++14	22 / 100	4003 ms	45984 KiB

plan

#include<bits/stdc++.h>
using namespace std;
int n,m,k,q;
bool is_npp[100000];
vector<pair<int,int>> edges[100000];
vector<int> sp(100000,INT_MAX);
//int furthest[50000][50000];
vector<int> npp;
void dijkstra()
{
    priority_queue<pair<int,int>> q;
    vector<int> dist(n+1,INT_MAX);
    for(int i=0;i<npp.size();i++)
    {
        q.push({0,npp[i]});
        dist[npp[i]]=0;
        sp[npp[i]]=0;
    }
    while(!q.empty())
    {
        int curr=q.top().second;
        int currDist=-q.top().first;
        q.pop();
        if(dist[curr]<currDist) continue;
        for(auto edge:edges[curr])
        {
            int next=edge.first;
            int weight=edge.second;
            if(dist[curr]+weight<dist[next])
            {
                dist[next]=dist[curr]+weight;
                sp[next]=min(sp[next],dist[next]);
                q.push({-dist[next],next});
            }
        }
    }
}
int dijkstra(int start,int finish)
{
    priority_queue<pair<int,int>> q;
    vector<int> away(n+1,0);
    q.push({sp[start],start});
    away[start]=sp[start];
    while(!q.empty())
    {
        int curr=q.top().second;
        int currAway=q.top().first;
        q.pop();
        if(away[curr]>currAway) continue;
        for(auto edge:edges[curr])
        {
            int next=edge.first;
            int len=sp[next];
            if(away[next]<min(len,away[curr]))
            {
                away[next]=min(len,away[curr]);
                q.push({away[next],next});
            }
        }
    }
    /*
    for(int i=1;i<=n;i++)
    {
        if(away[i]!=0) {furthest[start][i]=away[i];furthest[i][start]=away[i];}
    }*/
    return away[finish];
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>m;
    int ai,bi,wi;
    for(int i=0;i<m;i++)
    {
        cin>>ai>>bi>>wi;
        edges[ai].push_back({bi,wi});
        edges[bi].push_back({ai,wi});
    }
    cin>>k;
    for(int i=0;i<k;i++)
    {
        int npp1;
        cin>>npp1;
        is_npp[npp1]=true;
        npp.push_back(npp1);
       // dijkstra(npp);
    }
    dijkstra();
    cin>>q;
    int si,ti;
    for(int i=0;i<q;i++)
    {
        cin>>si>>ti;
        if(is_npp[si]==true || is_npp[ti]==true) {printf("0\n");continue;}
 //       if(furthest[si][ti]!=0) {printf("%d\n",furthest[si][ti]);continue;}
        printf("%d\n",dijkstra(si,ti));
    }
}

Compilation message (stderr)

plan.cpp: In function 'void dijkstra()':
plan.cpp:13:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   13 |     for(int i=0;i<npp.size();i++)
      |                 ~^~~~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 13
• Subtask 3: 12 / 12
• Subtask 4: 0 / 19
• Subtask 5: 0 / 46