ABC190 E - Magical Ornament （最短路+状压dp）

题意：

解法：

先对可相邻的数 建立无向边,
然后对k个关键点进行k次bfs,
计算出一个关键点到达另一个关键点需要的最小代价,
由于k<=17,做k次复杂度O(km),不会炸.

剩下的就是经典tsp问题:走遍k个点需要的最小代价,
令f[i][j]表示当前已经选取的集合为i,最后一个选的是j的最小代价,
状压dp一下即可.

code：

#include<bits/stdc++.h>
typedef long long ll;
#define int long long
using namespace std;
const int maxm=2e6+5;
vector<int>g[maxm];
map<int,int>mp;
int f[(1<<17)][17];
int dist[20][20];
int d[maxm];
int c[maxm];
int n,m,k;
void bfs(int st){
    for(int i=0;i<n;i++)d[i]=-1;
    queue<int>q;
    q.push(c[st]);
    d[c[st]]=0;
    while(q.size()){
        int x=q.front();q.pop();
        for(int v:g[x]){
            if(d[v]==-1){
                d[v]=d[x]+1;
                if(mp.count(v))dist[st][mp[v]]=d[v];
                q.push(v);
            }
        }
    }
}
void solve(){
    cin>>n>>m;
    for(int i=1;i<=m;i++){
        int a,b;cin>>a>>b;
        a--,b--;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    cin>>k;
    for(int i=0;i<k;i++){
        cin>>c[i];
        c[i]--;
        mp[c[i]]=i;
    }
    for(int i=0;i<k;i++){
        for(int j=0;j<k;j++){
            dist[i][j]=1e18;
        }
    }
    for(int i=0;i<k;i++){
        bfs(i);
    }
    for(int i=0;i<(1<<k);i++){
        for(int j=0;j<k;j++){
            f[i][j]=1e18;
        }
    }
    for(int j=0;j<k;j++){
        f[(1<<j)][j]=1;
    }
    for(int i=0;i<(1<<k);i++){
        for(int j=0;j<k;j++){
            if(f[i][j]==1e18)continue;
            for(int p=0;p<k;p++){
                if(!(i>>p&1)){
                    f[i|(1<<p)][p]=min(f[i|(1<<p)][p],f[i][j]+dist[j][p]);
                }
            }
        }
    }
    int ans=1e18;
    for(int j=0;j<k;j++){
        ans=min(ans,f[(1<<k)-1][j]);
    }
    if(ans==1e18)ans=-1;
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    int T=1;
    while(T--){
        solve();
    }
    return 0;
}

---