제출 #1344390

#제출 시각아이디문제언어결과실행 시간메모리
1344390WarinchaiCities (BOI16_cities)C++20
100 / 100
2662 ms51320 KiB
#include<bits/stdc++.h>
#define int long long
using namespace std;

vector<pair<int,int>>adj[100005];
int dis[35][100005];
int vis[100005];
int inf=1e18;
int n,k,m;

int32_t main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>k>>m;
    for(int i=0;i<(1<<k);i++)for(int j=1;j<=n;j++)dis[i][j]=inf;
    for(int i=0;i<k;i++){
        int x;cin>>x;
        dis[(1<<i)][x]=0;
    }
    for(int i=0;i<m;i++){
        int a,b,c;cin>>a>>b>>c;
        adj[a].push_back({b,c});
        adj[b].push_back({a,c});
    }
    //cerr<<"work\n";
    for(int i=0;i<(1<<k);i++){
        for(int j=i;j>0;j=(j-1)&i){
            for(int st=1;st<=n;st++){
                dis[i][st]=min(dis[i][st],dis[i^j][st]+dis[j][st]);
            }
        }
        priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>>pq;
        for(int j=1;j<=n;j++)pq.push({dis[i][j],j}),vis[j]=0;
        while(!pq.empty()){
            auto [c,x]=pq.top();
            pq.pop();
            if(vis[x])continue;
            vis[x]=1;
            dis[i][x]=c;
            for(auto [u,cc]:adj[x]){
                pq.push({c+cc,u});
            }
        }
    }
    int ans=inf;
    for(int i=1;i<=n;i++)ans=min(ans,dis[(1<<k)-1][i]);
    cout<<ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...