Magical Ornament（最短路,状压dp）|| 洛谷 P1433 吃奶酪（同类题）

题目链接：

E - Magical Ornament

吃奶酪 - 洛谷

思路：

1.Magical Ornament

稍加思索之后不难发现，任意两个要求的宝石之间的最短距离其实是固定的，所以这部分内容可以用bfs进行预处理。剩下就是图中k个点连通求最短路的问题了，典型的状压dp（dp[ i ][ j ]表示 i 状态下以第 j 个所求宝石为终点的最小长度，每个dp值都可以由【上一个状态的dp值 + 新的所求宝石和已有的所求宝石的最短距离】取最小算出。

----------------------------------------------------------------------------------------------------------------------------

2.吃奶酪

同样的思路，最短路状压dp模板题。

学习笔记：把double数组设置成inf，不能用0x3f3f3f3f，要用127。

比如：memset(dp, 127, sizeof(dp));

代码：

1.Magical Ornament

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
#include <string.h>
#include <map>
typedef long long ll;
using namespace std;

const int maxn = 1e5+5;
const int inf = 0x3f3f3f3f;
vector<int> g[maxn];
int dist[17][17]; // 所求宝石之间的距离
int dp[(1<<17)][17];
int d[maxn]; // 当前距离
int c[17];
int n, m, k;

void bfs(int st){
    for(int i=0; i<maxn; i++){
        d[i] = -1;
    }
    int x = c[st]; // x是起点的编号，st是起点在c中的下标
    d[x] = 0;
    queue<int> q;
    q.push(x);
    while(q.size()){
        int u = q.front(); q.pop();
        for(int v : g[u]){
            if(d[v] == -1){
                d[v] = d[u] + 1;
                for(int i=0; i<k; i++){
                    // 如果v是所求宝石，记录它和起点的距离（记录的时候要用c下标）
                    if(c[i] == v) dist[st][i] = d[v];
                }
                q.push(v); //入队列，进行下一轮bfs
            }
        }
    }
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    // 输入
    cin >> n >> m;
    for(int i=0; i<m; i++){
        int a, b;
        cin >> a >> b;
        g[a].push_back(b);
        g[b].push_back(a);
    }
    cin >> k;
    for(int i=0; i<k; i++){
        cin >> c[i];
    }

    //bfs预处理
    memset(dist, inf, sizeof(dist));
    for(int i=0; i<k; i++){
        bfs(i);
    }

    memset(dp, inf, sizeof(dp));
    //dp初始化
    for(int i=0; i<k; i++){
        dp[1<<i][i] = 1; //只有一个宝石，长度是1
    }
    //dp递推
    for(int mask=1; mask<(1<<k); mask++){ // 遍历所有状态
        for(int i=0; i<k; i++){ // 遍历子状态出发点
            if(!(mask & (1<<i))) continue; //没有这个点，不能从此出发，跳过
            for(int j=0; j<k; j++){ // 遍历下一个点，推出母状态
                if(mask & (1<<j)) continue; //已经有这个点，跳过
                dp[mask | (1<<j)][j] = min(dp[mask | (1<<j)][j], dp[mask][i] + dist[i][j]);
            }
        }
    }
    int ans = inf;
    for(int i=0; i<k; i++){
        ans = min(ans, dp[(1<<k)-1][i]);
    }
    if(ans == inf) ans = -1;
    cout << ans << endl;
}

2.吃奶酪

#include <bits/stdc++.h>
using namespace std;
const int maxn = 15;
int n;
double x[maxn], y[maxn];
double dp[(1<<maxn)][maxn]; //状态，终点，dp值表示最短距离

double getDist(double a, double b, double c, double d){
    return sqrt(pow(a-c,2) + pow(b-d, 2));
}
int main(){
    cin >> n;
    for(int i=0; i<n; i++){
        cin >> x[i] >> y[i];
    }
    memset(dp, 127, sizeof(dp)); //double数组设为inf要用127，不是0x3f3f3f3f
    for(int i=0; i<n; i++){
        dp[1<<i][i] = getDist(0, 0, x[i], y[i]); //dp初始化，计算起点到每个点的距离
    }
    for(int mask=1; mask<(1<<n); mask++){
        for(int i=0; i<n; i++){
            if(!mask & (1<<i)) continue; //枚举子状态出发点
            for(int j=0; j<n; j++){
                if(mask & (1<<j)) continue; //枚举母状态终点
                dp[mask | (1<<j)][j] = min(dp[mask | (1<<j)][j], dp[mask][i] + getDist(x[i],y[i],x[j],y[j]));
            }
        }
    }

    double ans = 0x3f3f3f3f;
    for(int i=0; i<n; i++){
        ans = min(ans, dp[(1<<n)-1][i]);
    }
    printf("%.2lf\n",ans);
}