Submission #113708

#	Time	Username	Problem	Language	Result	Execution time	Memory
113708		thebes	Cities (BOI16_cities)	C++14	74 / 100	6100 ms	200920 KiB

cities

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,ssse3,sse3,sse4,popcnt,avx,mmx,abm,tune=native")
#include <bits/stdc++.h>
using namespace std;

const int MN = 1e5+5;
typedef long long ll;
typedef pair<ll,ll> pii;
struct reeee{int n, m; ll d;};

bool ww[MN][1<<6];
ll n, m, k, i, x, y, w, vs[MN][1<<6], ans = 1LL<<60;
vector<pii> adj[MN]; vector<ll> heh;
struct pq{bool operator()(const reeee&i,const reeee&j){return i.d>j.d;}};
priority_queue<reeee,vector<reeee>,pq> q;

int main(){
    for(scanf("%lld%lld%lld",&n,&k,&m),i=1;i<=k;i++){
        scanf("%lld",&x);
        heh.push_back(x);
    }
    for(i=1;i<=m;i++){
        scanf("%lld%lld%lld",&x,&y,&w);
        adj[x].push_back({y,w});
        adj[y].push_back({x,w});
    }
    memset(vs,0x3f,sizeof(vs));
    for(i=0;i<heh.size();i++)
        q.push({heh[i], 1<<i, 0});
    while(q.size()){
        auto v = q.top(); q.pop();
        if(v.m==(1<<k)-1) ans = min(ans, v.d);
        if(ww[v.n][v.m]) continue;
        vs[v.n][v.m] = v.d;
        ww[v.n][v.m] = 1;
        for(int mm=0;mm<(1<<k);mm++){
            if(v.d+vs[v.n][mm]<vs[v.n][v.m|mm]){
                vs[v.n][v.m|mm]=v.d+vs[v.n][mm];
                q.push({v.n, v.m|mm, v.d+vs[v.n][mm]});
            }
        }
        for(auto w : adj[v.n]){
            if(v.d+w.second<vs[w.first][v.m]){
                vs[w.first][v.m]=v.d+w.second;
                q.push({w.first, v.m, v.d+w.second});
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}

Compilation message (stderr)

cities.cpp: In function 'int main()':
cities.cpp:29:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(i=0;i<heh.size();i++)
             ~^~~~~~~~~~~
cities.cpp:30:33: warning: narrowing conversion of 'heh.std::vector<long long int>::operator[](((std::vector<long long int>::size_type)i))' from '__gnu_cxx::__alloc_traits<std::allocator<long long int> >::value_type {aka long long int}' to 'int' inside { } [-Wnarrowing]
         q.push({heh[i], 1<<i, 0});
                                 ^
cities.cpp:46:27: warning: narrowing conversion of 'w.std::pair<long long int, long long int>::first' from 'long long int' to 'int' inside { } [-Wnarrowing]
                 q.push({w.first, v.m, v.d+w.second});
                         ~~^~~~~
cities.cpp:19:39: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(scanf("%lld%lld%lld",&n,&k,&m),i=1;i<=k;i++){
         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~
cities.cpp:20:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld",&x);
         ~~~~~^~~~~~~~~~~
cities.cpp:24:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld%lld%lld",&x,&y,&w);
         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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