Submission #69876

#	Time	Username	Problem	Language	Result	Execution time	Memory
69876		Bodo171	Cities (BOI16_cities)	C++14	74 / 100	6056 ms	172884 KiB

cities

#include <iostream>
#include <fstream>
#include <queue>
#include <vector>
#include <climits>
using namespace std;
const int nmax=100005;
vector< pair<int,long long> > v[nmax];
long long d[nmax][32];
bool E[nmax][32];
int norm[nmax],imp[10];
long long di,newd,ans;
int x,n,m,nod,k,i,j,y,z,stare,stare_noua,orig;
struct node
{
    long long D;
    int node,state;
};
struct cmp
{
    bool operator()(node unu,node doi)
    {
        return unu.D>doi.D;
    }
};
priority_queue <node,vector<node>,cmp > pq;
void dij()
{
    while(!pq.empty())
    {
        x=pq.top().node;di=pq.top().D;stare=pq.top().state;
        pq.pop();
        if(stare==(1<<k)-1)
        {
            ans=di;
            return;
        }
        if(!E[x][stare])
        {
            E[x][stare]=1;
            for(i=0;i<v[x].size();i++)
            {
                nod=v[x][i].first;newd=v[x][i].second+di;
                if(newd<d[nod][stare])
                {
                    d[nod][stare]=newd;
                    pq.push({newd,nod,stare});
                }
            }
            nod=x;orig=(((1<<k)-1)^stare);
            for(i=orig;i!=0;i=((i-1)&orig))
            {
                stare_noua=(stare|i);newd=1LL*d[x][stare]+d[x][i];
                if(newd<d[x][stare_noua])
                {
                    d[x][stare_noua]=newd;
                    pq.push({newd,x,stare_noua});
                }
            }
        }
    }
}
int main()
{
   // freopen("data.in","r",stdin);
    ios_base::sync_with_stdio(false);
    cin>>n>>k>>m;
    for(i=1;i<=k;i++)
    {
        cin>>imp[i];
        norm[imp[i]]=i;
    }
    for(i=1;i<=m;i++)
    {
        cin>>x>>y>>z;
        v[x].push_back({y,z});
        v[y].push_back({x,z});
    }
    for(i=1;i<=n;i++)
      for(j=1;j<(1<<k);j++)
         d[i][j]=1LL*1e17;
    for(i=1;i<=k;i++)
    {
        d[imp[i]][(1<<(i-1))]=0;
        pq.push({0,imp[i],(1<<(i-1))});
    }
    ans=LLONG_MAX;
    dij();
    cout<<ans;
    return 0;
}

Compilation message (stderr)

cities.cpp: In function 'void dij()':
cities.cpp:41:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
             for(i=0;i<v[x].size();i++)
                     ~^~~~~~~~~~~~

• Subtask 1: 22 / 22
• Subtask 2: 14 / 14
• Subtask 3: 15 / 15
• Subtask 4: 23 / 23
• Subtask 5: 0 / 26