Submission #1245797

#	Time	Username	Problem	Language	Result	Execution time	Memory
1245797		vuxxcode	Cities (BOI16_cities)	C++20	22 / 100	62 ms	34120 KiB

cities

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define eb emplace_back
#define pb push_back
#define fi first
#define se second
#define ii pair<int,int>
#define pl pair<ll,ll>
#define ve vector
#define vi ve<int>
#define vvi ve<vi>
#define vl ve<ll>
#define vii ve<ii>
#define all(x) x.begin(), x.end()
#define fo(i,a,b) for (int i=(a),_b=(b); i<=_b; ++i)
#define fd(i,a,b) for (int i=(a),_b=(b); i>=_b; --i)
#define lc p+1
#define rc p+(mid-l+1)*2
#define maxi(a, b) a = max(a, b)
#define mini(a, b) a = min(a, b)
#define bit(s, i) (((s) >> (i)) & 1)
#define tup3 array<int,3>
#define tup4 array<int,4>
#define matrix vvi
#define cntBit(x) (__builtin_popcountll(x))
#define SZ(x) int(x.size())
#define x1 _
#define x2 __
#define y1 _
#define y2 __

const int N = 1e5+5, M = 1e6+5;
const int inf = 1e9+10;
const int mod = 1e9+7;
const int LG = 20, base=11111, B = 500;
bool st_mem;

int n, k, m;
ve<ii> g[N];
ll d[1<<5][N];

void sol() {
    cin >> n >> k >> m;
    vi v(k);
    fo(i,0,k-1) cin >> v[i];
    fo(i,1,m) {
        int u, v, c; cin >> u >> v >> c;
        g[u].eb(v, c), g[v].eb(u, c);
    }
    struct node {
        int u, mask; ll d;
        bool operator < (const node &o) const {
            return d > o.d;
        }
    };
    priority_queue<node> q;
    memset(d,0x3f,sizeof(d));
    fo(i,1,n) d[0][i]=0;
    fo(i,0,k-1) {
        q.push({v[i],1<<i,0});
        d[1<<i][v[i]]=0;
    }
    ll ans=1e18;
    while (q.size()) {
        node t = q.top(); q.pop();
        int u = t.u, mask = t.mask; ll dd = t.d;
        if (d[mask][u]^dd) continue;
        if (mask == (1 << k) - 1) {
            mini(ans, dd);
        }
        fo(i,1,(1<<k)-1) if (d[u][i] < 1e16 && d[mask|i][u] > dd + d[i][u]) {
            d[mask|i][u] = dd + d[i][u];
            q.push({u,mask|i,dd+d[i][u]});
        }
        for (ii &x:g[u]) {
            int v=x.fi, c=x.se;
            fo(i,0,(1<<k)-1) if (d[i][v] < 1e16 && d[i|mask][v] > dd + c + d[i][v]) {
                d[i|mask][v] = dd + c + d[i][v];
                q.push({v,i|mask,dd + c + d[i][v]});
            }
        }
    }
    cout << ans;
}

bool en_mem;
signed main() {
    ios::sync_with_stdio(0); cin.tie(0);
    if(fopen("A.inp","r")) {
        freopen("A.inp","r",stdin);
        freopen("A.out","w",stdout);
    }
    int tc = 1; // cin >> tc;
    fo(i,1,tc) sol();
     // cerr << abs(&en_mem - &st_mem) / 1024.0 / 1024.0;
    return 0;
}

Compilation message (stderr)

cities.cpp: In function 'int main()':
cities.cpp:91:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   91 |         freopen("A.inp","r",stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~
cities.cpp:92:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   92 |         freopen("A.out","w",stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~

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