AtCoder Beginner Contest 190 E - Magical Ornament(图论+状压dp)

题意：
给出n种宝石，有m种关系，现在要在一排放置一些宝石，只有存在关系的宝石才能相邻，并且要求必须放置c数组里的宝石，问最少要放多少宝石？

题解：
这题比F题难是我没想到的。。。
其实将题目转化一下就是：将所给的关系建边，然后在图中找到一条最短路径使得能包含所需所有点。
首先我们需要求出必须放置的点之间两两的最短路径。因为k很小，所以我们可以进行k次bfs，求出两点间的最短距离。
然后就是一个状压dp的问题了。设dp[i][j]表示状态 为 i 最后到达的点为 j 的最小路径。状态 i 变为二进制，第 i 位为1表示 c[i] 已经经过。那么可以推出转移方程:
dp[next_i][next_j]=min(dp[next_i][next_j],dp[i][j]+dis[j][next_j])。
最后枚举最后到达的点取min即可。

代码：

#include<bits/stdc++.h>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int,int>pii;
const int MAXN=1e5+200;
const int mod=77797;
const int inf=0x3f3f3f3f;
int n,m;
int k;
int c[20];
std::vector<int> v[MAXN];
ll dis[20][20];
int vis[MAXN];
ll d[MAXN];
ll dp[(1<<18)][20];
void bfs(int s)
{
    d[s]=0;
    dis[vis[s]][vis[s]]=0;
    queue<int >q;
    q.push(s);
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        int len=v[now].size();
        for(int i=0;i<len;i++)
        {
            if(d[v[now][i]]==inf)
            {
                d[v[now][i]]=d[now]+1;
                q.push(v[now][i]);
                if(vis[v[now][i]]) 
                {
                    //cout<<s<<" "<<v[now][i]<<endl;
                    dis[vis[s]][vis[v[now][i]]]=d[v[now][i]];
                }
            }
        }
    }
}
int main()
{
    memset(dis,inf,sizeof dis);
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int uu,vv;
        cin>>uu>>vv;
        v[uu].push_back(vv);
        v[vv].push_back(uu);
    }
    cin>>k;
    for(int i=1;i<=k;i++)
    {
        cin>>c[i];
        vis[c[i]]=i;
    }
    for(int i=1;i<=k;i++)
    {
        for(int j=1;j<=n;j++)
        {
            d[j]=inf;
        }
        bfs(c[i]);
    }
    /*for(int i=1;i<=k;i++)
    {
        for(int j=1;j<=k;j++)
        {
            cout<<c[i]<<" "<<c[j]<<" "<<dis[c[i]][c[j]]<<endl;
        }
    }*/
    memset(dp,inf,sizeof dp);
    for(int i=1;i<=k;i++)
    {
        dp[1<<(i-1)][i]=1;
    }
    for(int i=1;i<=(1<<k)-1;i++)
    {
        for(int j=1;j<=k;j++)
        {
            if(dp[i][j]==inf) continue;
            for(int z=1;z<=k;z++)
            {
                if(z==j) continue;
                if((1<<(z-1))&i) continue;
                dp[i|(1<<(z-1))][z]=min(dp[i|(1<<(z-1))][z],dp[i][j]+dis[j][z]);
            }
        }
    }
    ll ans=inf;
    for(int i=1;i<=k;i++)
    {
        //cout<<dp[(1<<k)-1][i]<<endl;
        ans=min(ans,dp[(1<<k)-1][i]);
    }
    if(ans==inf)printf("-1\n");
    else printf("%lld\n",ans);
}